题目链接:https://www.patest.cn/contests/gplt/L2-010
布置宴席最微妙的事情,就是给前来参宴的各位宾客安排座位。无论如何,总不能把两个死对头排到同一张宴会桌旁!这个艰巨任务现在就交给你,对任何一对客人,请编写程序告诉主人他们是否能被安排同席。
输入格式:
输入第一行给出3个正整数:N(<= 100),即前来参宴的宾客总人数,则这些人从1到N编号;M为已知两两宾客之间的关系数;K为查询的条数。随后M行,每行给出一对宾客之间的关系,格式为:“宾客1 宾客2 关系”,其中“关系”为1表示是朋友,-1表示是死对头。注意两个人不可能既是朋友又是敌人。最后K行,每行给出一对需要查询的宾客编号。
这里假设朋友的朋友也是朋友。但敌人的敌人并不一定就是朋友,朋友的敌人也不一定是敌人。只有单纯直接的敌对关系才是绝对不能同席的。
输出格式:
对每个查询输出一行结果:如果两位宾客之间是朋友,且没有敌对关系,则输出“No problem”;如果他们之间并不是朋友,但也不敌对,则输出“OK”;如果他们之间有敌对,然而也有共同的朋友,则输出“OK but...”;如果他们之间只有敌对关系,则输出“No way”。
输入样例:7 8 4 5 6 1 2 7 -1 1 3 1 3 4 1 6 7 -1 1 2 1 1 4 1 2 3 -1 3 4 5 7 2 3 7 2输出样例:
No problem OK OK but... No way
这道题入手时首先想到两种方法,prim和并查集,很明显prim是时间复杂度更高,交上去说有一个样例超时了,然后改用了并查集,主要是题要读清楚,题目的意思是:朋友的朋友是我的朋友,但是敌人的敌人不一定是我的敌人,一开始我就搞错了,觉得敌人的敌人还是我的敌人。。。就一直wa。。,上代码:
或许peime也能过。。我当时读错题了多写了一个prim,就超时了,其实就一个就行了,或许能过,我没试,有兴趣的可以试一下
#include<stdio.h>
#include<string.h>
#include<math.h>
//#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
#define da 0x3f3f3f3f
#define xiao -0x3f3f3f3f
#define clean(a,b) memset(a,b,sizeof(a))// 雷打不动的头文件
struct ac{
int a,b,relation;
}shuzu[11000];
int perf[110];//,perd[110]; //直连的朋友
int mapd[110][110]; //直接联系的敌对关系
int findf(int x) //找一个共同的朋友
{
while(perf[x]!=x)
x=perf[x];
return x;
}
/*int findd(int x)
{
while(perd[x]!=x)
x=perd[x];
return x;
}*/
void mixf(int a,int b) //朋友的朋友还是我的朋友
{
int aa=findf(a);
int bb=findf(b);
perf[aa]=perf[bb];
}
/*void mixd(int a,int b) //之前的敌人的敌人还是我的敌人(错的。。)忽略即可。。
{
int aa=findd(a);
int bb=findd(b);
perd[aa]=perd[bb];
}*/
int main()
{
int n,m,k;
cin>>n>>m>>k;
int i,j;
for(i=0;i<110;++i)
{
perf[i]=i; //自己先是自己的朋友
//perd[i]=i;
}
clean(mapd,0); //所有的人之间都没有敌对关系
for(i=0;i<m;++i)
{
cin>>shuzu[i].a>>shuzu[i].b>>shuzu[i].relation; //输入人与人和关系
if(shuzu[i].relation==1) //朋友关系,找到共同的朋友
mixf(shuzu[i].a,shuzu[i].b);
else
{
mapd[shuzu[i].a][shuzu[i].b]=-1; //敌对关系,互相敌对,但是不诛连
mapd[shuzu[i].b][shuzu[i].a]=-1; //直接的敌对关系是相互的
}//mixd(shuzu[i].a,shuzu[i].b);
}
while(k--) //查找k组
{
int a,b;
cin>>a>>b;
if(findf(perf[a])==findf(perf[b])) //判断有没有相同的朋友
{
if(mapd[a][b]==-1)//(findd(perd[a])==findd(perd[b])) //直接的敌对关系
cout<<"OK but..."<<endl;
else
cout<<"No problem"<<endl;
}
else
{
if(mapd[a][b]==-1)//(findd(perd[a])==findd(perd[b])) //直接的敌对关系
cout<<"No way"<<endl;
else
cout<<"OK"<<endl;
}
}
}
并查集