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[],is[][],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=;i<=n;++i)
         root[i]=i;
     int a,b,c;
     while(m--){
         scanf("%d%d%d",&a,&b,&c);
         if(c==){
             int ar=getr(a),br=getr(b);
             if(ar!=br)
                 root[br]=ar;
         }
         else
             is[a][b]=is[b][a]=;
     }
     while(k--){
         scanf("%d%d",&a,&b);
         int ar=getr(a),br=getr(b);
         if(ar==br&&!is[a][b]) printf("No problem\n");
         else if(ar==br&&is[a][b]) printf("OK but...\n");
         else if(ar!=br&&!is[a][b]) printf("OK\n");
         else printf("No way\n");
     }
     return ;
 }