L2-010 排座位（并查集）

Posted 2022-03-30 MangataTS

题目链接

https://pintia.cn/problem-sets/994805046380707840/problems/994805066135879680

思路

对于朋友和朋友之间的关系我们通过并查集维护即可，然后对于敌人之间的关系，我们只需要用一个二维数组或者使用map就能维护，因为只有直接敌对关系才是敌人，然后对于每一次询问我们就根据他们俩的朋友关系、敌对关系分类讨论即可

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define mod 1000000007
#define endl "\\n"
#define PII pair<int,int>
#define INF 0x3f3f3f3f

const int N = 1e2+10;

int fa[N],n,m,k;
bool enmy[N][N];
int find(int x)
	return x==fa[x]?x:find(fa[x]);


int main()

	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>n>>m>>k;
	for(int i = 1;i <= n; ++i) fa[i] = i;
	int u,v,w;
	for(int i = 1;i <= m; ++i) 
		cin>>u>>v>>w;
		if(w + 1) 
			u = find(u);
			v = find(v);
			fa[v] = u;
		 else 
			enmy[u][v] = enmy[v][u] = true;
		
	
	for(int i = 1;i <= k; ++i) 
		cin>>u>>v;
		if(enmy[u][v]) 
			if(find(u) == find(v)) cout<<"OK but..."<<endl;
			else cout<<"No way"<<endl;
		 else 
			if(find(u) == find(v))
				cout<<"No problem"<<endl;
			else  cout<<"OK"<<endl;
		
	
	return 0;