PATL2-010. 排座位-并查集

    xiaoxiao2021-03-25  148

    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 ; }
    转载请注明原文地址: https://ju.6miu.com/read-3019.html

    最新回复(0)