L2-010. 排座位

L2-010. 排座位

题目链接：https://www.patest.cn/contests/gplt/L2-010

并查集

相关题目：L2-007. 家庭房产，L3-003. 社交集群

下午打的时候什么都没用，直接上一个矩阵，没有考虑“甲是a的朋友，a是b的朋友，b是乙的朋友”这种多个人关系的情况，华丽WA。出来问了学长，用的是并查集，用一个矩阵来表示两人之间的敌对关系(写的时候还是WA了好多次，最后才发现是"No way"的"w"打成了大写QAQ)

代码如下：

#include<cstdio>
using namespace std;
bool enemy[105][105];
int pre[105];
int Find(int a){
    if(a!=pre[a])pre[a]=Find(pre[a]);
    return pre[a];
}
void uion(int a,int b){
    int t1=Find(a),t2=Find(b);
    if(t1!=t2)pre[t1]=t2;
}
void make(int n){
    for(int i=1;i<=n;i++)pre[i]=i;
}
int main(void){
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    make(n);
    while(m--){
        int a,b,g;
        scanf("%d%d%d",&a,&b,&g);
        if(g==1)uion(a,b);
        else enemy[a][b]=enemy[b][a]=1; //a和b是敌人
    }
    while(k--){
        int a,b;
        scanf("%d%d",&a,&b);
        int ta=Find(a),tb=Find(b);
        if(enemy[a][b]){
            if(ta==tb)printf("OK but...\n");
            else printf("No way\n");
        }else{
            if(ta==tb)printf("No problem\n");
            else printf("OK\n");
        }
    }
    return 0;
}