PATL2-010. 排座位-并查集

L2-010. 排座位

题目：
输入第一行给出3个正整数：N（<= 100），即前来参宴的宾客总人数，则这些人从1到N编号；M为已知两两宾客之间的关系数；K为查询的条数。随后M行，每行给出一对宾客之间的关系，格式为：“宾客1 宾客2 关系”，其中“关系”为1表示是朋友，-1表示是死对头。注意两个人不可能既是朋友又是敌人。最后K行，每行给出一对需要查询的宾客编号。
这里假设朋友的朋友也是朋友。但敌人的敌人并不一定就是朋友，朋友的敌人也不一定是敌人。只有单纯直接的敌对关系才是绝对不能同席的。

对每个查询输出一行结果：如果两位宾客之间是朋友，且没有敌对关系，则输出“No problem”；如果他们之间并不是朋友，但也不敌对，则输出“OK”；如果他们之间有敌对，然而也有共同的朋友，则输出“OK but…”；如果他们之间只有敌对关系，则输出“No way”。

---

题解：
简单的并查集，通过并查集完成“朋友的朋友也是朋友这句话”，由于敌对关系无传递性，故通过一个二位数组存储两两之间是否有敌对，之后通过条件选择判断按要求输出即可。

代码：

#include <stdio.h>
#include <iostream>
#include <string.h>
using namespace std;
#define MAX 110
int pf[MAX] ,pa[MAX][MAX]={0};
int Find(int arr[],int x)
{
    while(x != arr[x])
        x = arr[x];
    return x ;
}
void join(int a,int b)
{
    int x = pf[Find(pf,a)] ;
    int y = pf[Find(pf,b)] ;
    if(x != y)
    {
        if(x > y) pf[x] = y ;
        else pf[y] = x ;
    }
}
int main()
{
    int N , M , K ;
    int t1 , t2 , cc;
    scanf("%d%d%d" , &N,&M,&K);
    for(int i = 0 ; i < MAX ; i ++)
    {
        pf[i] = i ;
    }
    for(int i = 0 ; i < M ; i ++)
    {
        scanf("%d%d%d" , &t1,&t2,&cc);
        if(cc == 1)
        {
            join(t1 , t2) ;
        }
        else
        {
            pa[t1][t2] = 1 ;
            pa[t2][t1] = 1 ;
        }
    }
    for(int i = 0 ; i <K ; i ++)
    {
        scanf("%d%d" , &t1 , &t2);
        int f1 = Find(pf,t1),f2=Find(pf,t2);
        //cout << f1 <<" "<<f2<<endl;
        int a1 = pa[t1][t2];
        //cout << a1 <<endl;
        if(f1==f2&&a1==0) printf("No problem\n");
        else if(f1==f2&&a1==1) printf("OK but...\n");
        else if(f1!=f2&&a1==0) printf("OK\n");
        else if(f1!=f2&&a1==1) printf("No way\n");
    }
    return 0 ;
}