一开始我用的是搜索法,额,代码有点长,但是不容易错。
然后当时想的是不是可以用一个二维数组将它们的关系存起来呢,然后比完赛以后,敲了下这个方法,还是发现了自己一些思维上的问题。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define maxn 110
int vis[maxn][maxn];
int dis[maxn];
int findf(int kk)
{
if(dis[kk]==kk)
return kk;
else return kk=findf(dis[kk]);
}
int main()
{
memset(vis,0,sizeof(vis));
int n,m,k;
for(int i=0;i<maxn;i++)
dis[i]=i;
scanf("%d %d %d",&n,&m,&k);
for(int i=0;i<m;i++)
{
int a,b,w;
scanf("%d %d %d",&a,&b,&w);
vis[a][b]=w;
vis[b][a]=w;
if(w==1) {
a=findf(a),b=findf(b);
if(a!=b) dis[a]=b;
}
}//开始的时候并没有用并查集,因为忽略了有可能是间接朋友
for(int i=0;i<k;i++)
{
int a,b;
scanf("%d %d",&a,&b);int root1=findf(a),root2=findf(b);
if(vis[a][b]!=-1&&root1==root2) printf("No problem\n");//当root1==root2的时候也是可能不会是NP,可能a,b是敌人,但是他们有共同的朋友
else if(vis[a][b]==0) printf("OK\n");
else if(vis[a][b]==-1){
if(root1==root2)
printf("OK but...\n");
else printf("No way\n");
}
}
return 0;
}