题目链接:题目详情 - L2-010 排座位 (pintia.cn)
样例输入:
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
分析:题目中给出朋友关系具有传递性质,那么我们显然可以用并查集来维护朋友关系,位于同一个集合中的人互为朋友,但是可以发现两个互为敌人的人也可能位于同一个集合,但是敌对关系并不具有传递性质,所以我们可以直接用一个数组来直接维护,那么在一个集合中但是互为敌人的人我们可以只要优先考虑敌对关系即可。最后直接根据题目所给进行输出即可。
细节见代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<map>
#include<queue>
#include<vector>
#include<cmath>
using namespace std;
const int N=103;
int fu[N];
bool vis[N][N];//vis[i][j]=true代表i和j是敌人
int find(int x)
{
if(x!=fu[x]) return fu[x]=find(fu[x]);
return x;
}
int main()
{
int n,m,k;
cin>>n>>m>>k;
for(int i=1;i<=n;i++)
fu[i]=i;
for(int i=1;i<=m;i++)
{
int x,y,op;
scanf("%d%d%d",&x,&y,&op);
if(op==1)
{
int fx=find(x),fy=find(y);
if(fx==fy) continue;
fu[fy]=fx;
}
else
vis[x][y]=vis[y][x]=true;
}
for(int i=1;i<=k;i++)
{
int x,y;
scanf("%d%d",&x,&y);
int fx=find(x),fy=find(y);
if(fx==fy&&!vis[x][y]) puts("No problem");
else if(fx!=fy&&!vis[x][y]) puts("OK");
else if(fx==fy&&vis[x][y]) puts("OK but...");
else puts("No way");
}
return 0;
}