首页 > 代码库 > poj1703Find them, Catch them(并查集以及路径压缩)
poj1703Find them, Catch them(并查集以及路径压缩)
1 /* 2 题目大意:有两个不同的黑帮,开始的时候不清楚每个人是属于哪个的! 3 执行两个操作 4 A a, b回答a, b两个人是否在同一帮派,或者不确定 5 D a, b表示a, b两个人不在同一个帮派 6 7 思路:利用并查集将相同帮派的人合并到一起! 8 a ,b 不在同一个城市,那么 a, 和mark[b]就在同一个城市, 9 b 和 mark[a]就在同一个城市!10 */11 #include<iostream>12 #include<cstring>13 #include<cstdio>14 #define M 10000515 using namespace std; 16 int f[M];17 int mark[M];//mark[i]表示 i 与 mark[i]不在同一个黑帮 18 int rank[M];//为不同的集合的父亲节点记录其孩子机节点个数,在合并集合的时候,将19 int n, m; //rank 值小的集合连接到 rank 值大的集合上去! 这样在路径压缩的时候省去不少时间 20 21 void init(){22 23 memset(mark, 0, sizeof(int)*(n+1));24 memset(rank, 0, sizeof(int)*(n+1));25 for(int i=1; i<=n; ++i)26 f[i]=i;27 }28 29 int getFather(int x){30 return x==f[x] ? x : f[x]=getFather(f[x]); 31 }32 33 void Union(int a, int b){34 int fa=getFather(a), fb=getFather(b);35 if(fa!=fb){36 if(rank[fa]>rank[fb]){ 37 f[fb]=fa;38 rank[fa]+=rank[fb]+1;39 }40 else{41 f[fa]=fb;42 rank[fb]+=rank[fa]+1;43 }44 }45 } 46 47 int main(){48 int t;49 char cmd[3];50 scanf("%d", &t);51 while(t--){52 scanf("%d%d", &n, &m);53 init();54 while(m--){55 int u, v;56 scanf("%s%d%d", cmd, &u, &v);57 if(cmd[0]==‘A‘){58 int fu=getFather(u), fv=getFather(v);59 if(fu==fv)60 printf("In the same gang.\n");61 else if(fu==getFather(mark[v]))62 printf("In different gangs.\n"); 63 else64 printf("Not sure yet.\n");65 }66 else{67 if(!mark[u] && !mark[v]){68 mark[u]=v;69 mark[v]=u;70 } 71 else if(!mark[u]){72 mark[u]=v;73 Union(u, mark[v]);74 }75 else if(!mark[v]){76 mark[v]=u;77 Union(v, mark[u]);78 }79 else {80 Union(u, mark[v]);81 Union(v, mark[u]);82 }83 }84 }85 } 86 return 0;87 }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。