pku 1703(种类并查集)

题目链接：http://poj.org/problem?id=1703

思路;个人觉得本质上还是和带权并查集一样的，只不过多了一个MOD操作，然后就是向量关系图稍微改动一下就变成种类并查集了，对于本题，我们可以用一个kind数组来表示是否属于同一类，其中kind[x]==0表示不是同一类，kind[x]==1表示属于同一类，这样我们就可以得到向量关系式了(若r1=Find(u),r2=Find(v),并且parent[r1]=r2,那么就有kind[v]+1==kind[u]+kind[r1])然后变形后对2取余就可以了，即kind[r1]=(kind[v]-kind[u]+1)%2，这是Union部分，还有路径压缩的时候注意更新一下kind[]就ok了。

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 using namespace std;
 5 #define MAXN 100010
 6 int parent[MAXN];
 7 int kind[MAXN];
 8 //0表示不属于同一类，1表示属于同一类
 9 int n,m;
10 
11 void Initiate()
12 {
13     for(int i=1;i<=n;i++){
14         parent[i]=i;
15     }
16     memset(kind,0,(n+2)*sizeof(kind[0]));
17 }
18 
19 int Find(int x)
20 {
21     if(x==parent[x]){
22         return parent[x];
23     }
24     int tmp=Find(parent[x]);
25     kind[x]=(kind[x]+kind[parent[x]])%2;
26     return parent[x]=tmp;
27 }
28 
29 void Union(int u,int v)
30 {
31     int r1=Find(u),r2=Find(v);
32     if(r1==r2)return ;
33     parent[r1]=r2;
34     kind[r1]=(kind[v]-kind[u]+1)%2;
35 }
36 
37 int main()
38 {
39    // freopen("1.txt","r",stdin);
40     int _case,a,b;
41     char str[4];
42     scanf("%d",&_case);
43     while(_case--){
44         scanf("%d%d",&n,&m);
45         Initiate();
46         while(m--){
47             scanf("%s%d%d",str,&a,&b);
48             if(str[0]=='D'){
49                 Union(a,b);
50             }else {
51                 int r1=Find(a),r2=Find(b);
52                 if(r1!=r2){
53                     puts("Not sure yet.");
54                 }else if(kind[a]==kind[b]){
55                     puts("In the same gang.");
56                 }else {
57                     puts("In different gangs.");
58                 }
59             }
60         }
61     }
62     return 0;
63 }

View Code