排座位

来源:互联网 发布:openpose python编译 编辑:程序博客网 时间:2024/04/28 18:28

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

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>#include <queue>using namespace std;#define maxn 110int 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;}
0 0
原创粉丝点击