poj1703 Find them, Catch them(种类并查集
题目地址:http://poj.org/problem?id=1703
题目大意:警察抓了n个坏蛋,这些坏蛋分别属于龙帮或蛇帮。输入m个语句,A x y询问x和y的关系(在一个帮派,不在,不能确定),D x y表示x和y不在一个帮派。
思路:种类并查集。维护一个数组rel[x]表示x和根节点的关系(0表示在一个帮派,1表示不在),初始全为0(自己和自己在一个帮派),更新rel用偏移量(rel[x]表示px->x,D x y表示x->y=1)。由于2要特判,输入里分三类:1.n==2&&opr=='A' 2.opr=='D' 3.其他【由于至少有两个帮派的人,当n==2是两个人一定分别是两个帮派的,输出为different】【初始化时记得n从1到n】
1 #include <cstdio> 2 3 using namespace std; 4 5 const int N = 100000+5; 6 int fa[N], rel[N]; 7 8 void init(int n) 9 { 10 for(int i = 0; i <= n; i++) 11 { 12 fa[i] = i; 13 rel[i] = 0; 14 } 15 } 16 17 //注意下查找时更新rel[]里那个px 18 int found(int x) 19 { 20 if(fa[x] == x) 21 return x; 22 int px = fa[x]; 23 fa[x] = found(fa[x]); 24 rel[x] = (rel[px]+rel[x])%2; 25 return fa[x]; 26 } 27 28 void unite(int x, int y) 29 { 30 int px = found(x); 31 int py = found(y); 32 if(px != py) 33 { 34 fa[py] = px; 35 rel[py] = (rel[x]+1+2-rel[y])%2; 36 } 37 } 38 39 int main() 40 { 41 int n, m, t, a, b; 42 char opr; 43 scanf("%d", &t); 44 while(t--) 45 { 46 scanf("%d%d", &n, &m); 47 init(n); 48 for(int i = 0; i < m; i++) 49 { 50 getchar(); 51 scanf("%c%d%d", &opr, &a, &b); 52 if(n == 2 && opr =='A')//2要特判。分类一共是A2、其他A、D 53 { 54 printf("In different gangs.\n"); 55 } 56 else if(opr == 'D') 57 unite(a, b); 58 else 59 { 60 int pa = found(a); 61 int pb = found(b); 62 if(pa != pb) 63 printf("Not sure yet.\n"); 64 else if(rel[a] == rel[b]) 65 printf("In the same gang.\n"); 66 else 67 printf("In different gangs.\n"); 68 } 69 } 70 } 71 return 0; 72 }
【特判2wa了好久!!!!!!!!!!!!!!!!!!!初始化函数也写错了。。刚开始写成<n了!!!!!!!!!!!】