排座位

一开始我用的是搜索法，额，代码有点长，但是不容易错。
然后当时想的是不是可以用一个二维数组将它们的关系存起来呢，然后比完赛以后，敲了下这个方法，还是发现了自己一些思维上的问题。

#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;
}