pta l2-10（排座位）

题目链接：https://pintia.cn/problem-sets/994805046380707840/problems/994805066135879680

题意：给宴席排座位，有n个人，m个关系，k组询问，1表示为朋友，-1表示为敌人。询问时，两人为朋友而非敌人输出No problem，两人既不为敌人也不为朋友输出OK，两人为敌人但有共同的朋友输出OK but...，两人只有敌人关系输出No way。

思路：并查集，题目看着很绕，其实仔细看就会发现很简单。我们把有朋友关系的人并起来，但这个集合里也可能有敌人关系，然后用二维数组is[i][j]记录i,j是否是敌人，即两个人之间可能有4种关系：

1. 有共同祖先，不敌对：No problem;
2. 有共同祖先，敌对：OK but...
3. 没有共同的祖先，不敌对：OK
4. 没有共同的祖先，敌对：No way

AC代码：

#include<bits/stdc++.h>
using namespace std;

int root[105],is[105][105],n,m,k;

int getr(int kk){
    if(root[kk]==kk) return kk;
    else return root[kk]=getr(root[kk]);
}

int main(){
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;++i)
        root[i]=i;
    int a,b,c;
    while(m--){
        scanf("%d%d%d",&a,&b,&c);
        if(c==1){
            int ar=getr(a),br=getr(b);
            if(ar!=br)
                root[br]=ar;
        }
        else
            is[a][b]=is[b][a]=1;
    }
    while(k--){
        scanf("%d%d",&a,&b);
        int ar=getr(a),br=getr(b);
        if(ar==br&&!is[a][b]) printf("No problem
");
        else if(ar==br&&is[a][b]) printf("OK but...
");
        else if(ar!=br&&!is[a][b]) printf("OK
");
        else printf("No way
");
    }
    return 0;
}