POJ 1703 && poj 1182【典型并查集】

http://poj.org/problem?id=1703

理解：一个数组用来保存每个节点之间的关系：

1、 father()根据子节点与父亲节点的关系和父节点与爷爷节点的关系，推导子节点与爷爷节点的关系



2.Union（）联合两棵树的时候也要更新两棵树的根的关系



题意 ：有两个不同的帮派，每个帮派至少有一个人。 判断两个人是否属于同一个帮派。

有 T 组测试数据。

给你 N 个人，编号从 1 到 N，操作 M 次。

每次操作输入一个字符和两个数 x ,y

如果字符为 A 则判断 x 和 y 是否属于同一个帮派，并且输出结果。

如果字符为 D 则明确告诉你 x 和 y 是属于不同帮派的。

算法：经典并查集的应用。

PS：以前接触的并查集都是让我们判断是否属于同一个连通分量，但这道题却让你判断是否属于同一类。

开始小纠结了下，如果我昨天没有纠结清楚 POJ 1182 食物链 那题想必肯定是做不出来的，其实想清楚了，这道题就是
食物链

那题的简单版本。


思路：除了像普通的并查集定义一个 p[] 记录父亲节点外，还定义一个 r[] 记录当前点与其所属的连通分量的根节点的关系。

r[] = 0 表示属于同一个帮派； r[] = 1表示与其根节点属于不同的帮派。

开始时初始化自己是自己的父亲 p[x] = x，自己与自己属于同一类 r[x] = 0.

一旦输入 D 断定 x 和 y 属于不同集合后，就连接 x 和 y 所在的树，同时更新 r[]

一旦输入 A

如果 find(x) 不等于 find(y) 说明还没有判断过 x 与 y 直接输出关系不确定即可

Not sure yet.

如果find(x)等于 find(y) ，但是他们的r不等，说明属于不同帮派，输出In different gangs.

如果他们的r相等，说明属于同一个帮派，则输出In the same gang

注意：1.find()函数寻找根节点的时候要不断的更新 r

根据子节点与父亲节点的关系和父节点与爷爷节点的关系，推导子节点与爷爷节点的关系

如果 a 和 b 的关系是 r1, b 和 c 的关系是 r2,

那么 a 和 c 的关系就是 (r1+r2)%2 . PS:因为只用两种情况所以对 2 取模。

如果实在不好理解，那么我们就枚举推理一下，共有 2*2 = 4种情况：

（a, b） (b, c)  (a, c)  (r1+r2)%2

0	 0       0        0    a 和 b是同类 ， b 和 c 是同类， 所以 a 和 c 也是同类

0      1       1        1    a 和 b是同类 ， b 和 c 是异类， 所以 a 和 c 也是异类

1      0       1        1    a 和 b是异类 ， b 和 c 是同类， 所以 a 和 c 是异类

1      1       0        0    a 和 b是异类 ， b 和 c 是异类， 所以 a 和 c 是同类

2.Union（）联合两棵树的时候也要更新两棵树的根的关系

定义：fx 为 x的根节点， fy 为 y 的根节点

联合时，使得 p[fx] = fy; 同时也要寻找 fx 与 fy 的关系。关系为：（r[x]+r[y]+1）%2

如何证明？

fx 与 x 的关系是 r[x],

x 与 y 的关系是 1 （因为确定是不同类，才联合的）,

y 与 fy 关系是 r[y],模 2 是因为只有两种关系

所以又上面的一点所推出的定理可以证明 fx 与 fy 的关系是： （r[x]+r[y]+1）%2

#include<cstdio>

const int maxn = 100000+10;

int p[maxn]; //存父亲节点
int r[maxn]; //存与根节点的关系，0 代表同类， 1代表不同类

int find(int x) //找根节点
{
if(x == p[x]) return x;

int t = p[x]; //记录父亲节点 方便下面更新r[]
p[x] = find(p[x]);
r[x] = (r[x]+r[t])%2; //根据子节点与父亲节点的关系和父节点与爷爷节点的关系，推导子节点与爷爷节点的关系
return p[x]; //容易忘记
}

void Union(int x, int y)
{
int fx = find(x); //x所在集合的根节点
int fy = find(y);

p[fx] = fy; //合并
r[fx] = (r[x]+1+r[y])%2; //fx与x关系 + x与y的关系 + y与fy的关系 = fx与fy的关系
}
void set(int n)
{
for(int x = 1; x <= n; x++)
{
p[x] = x; //自己是自己的父节点
r[x] = 0; //自己和自己属于同一类
}
}

int main()
{
int T;
int n, m;
scanf("%d", &T);
while(T--)
{
scanf("%d%d%*c", &n, &m);
set(n);

char c;
int x, y;
while(m--)
{
scanf("%c%d%d%*c", &c, &x, &y); //注意输入
//printf("%c\n", c);
if(c == 'A')
{
if(find(x) == find(y)) //如果根节点相同，则表示能判断关系
{
if(r[x] != r[y]) printf("In different gangs.\n");
else printf("In the same gang.\n");
}
else printf("Not sure yet.\n");
}
else if(c == 'D')
{
Union(x, y);
}
}
}
return 0;
}

poj 1182

http://poj.org/problem?id=1182

关键词：并查集 相对关系

思路：（用一个并查集就够了，同时对每个节点保持其到根结点的相对类别偏移量）

1.father[x]表示x的根结点。rank[x]表示father[x]与x的关系。rank[x]
== 0 表示father[x]与x同类；1表示father[x]吃x；2表示x吃father[x]。

2.怎样划分一个集合呢？

注意，这里不是根据x与father[x]是否是同类来划分。而是根据“x与father[x]能否确定两者之间的关系”来划分，若能确定x与father[x]的关系，则它们同属一个集合。

3.怎样判断一句话是不是假话？

假设已读入 D , X , Y , 先利用find_set()函数得到X , Y 所在集合的代表元素 xf ,yf ,若它们在同一集合（即 xf
== yf ）则可以判断这句话的真伪（ 据 2. ）.

若 D == 1 而 rank[X] != rank[Y] 则此话为假。（D
== 1 表示X与Y为同类，而从rank[X] != rank[Y]可以推出 X 与 Y 不同类。矛盾。）

若 D == 2 而 rank[X] == rank[Y] （X 与Y为同类）或者　rank[X]
== ( rank[Y] + 1 ) % 3 （Y吃X ）则此话为假。

4.上个问题中 r[X] == ( r[Y] + 1 ) % 3这个式子怎样推来？假设有Y吃X，那么r[X]和r[Y]的值是怎样的？

我们来列举一下: r[X] = 0 && r[Y] = 2

r[X] = 1 && r[Y] = 0

r[X] = 2 && r[Y] = 1

稍微观察一下就知道r[X] = ( r[Y] + 1 ) % 3;事实上，对于上个问题有更一般的判断方法：

若 ( r[Y] - r[X] + 3 ) % 3 != D - 1 ，则此话为假。（来自poj 1182中的Discuss ）

5、注意事项：

A、我们用x--r-->y表示x和y之间的关系是r，比如x--1--y代表x吃y。现在，若已知x--r1-->y，y--r2-->z，如何求x--?-->z？，于是我们不难发现，x--(r1+r2)%3-->z。这个就是在Find_Set（int
x）函数中用到的更新x与father[X]的关系

B、当D X Y时，则应合并X的根节点和Y的根节点，同时修改各自的rank。那么问题来了，合并了之后，被合并的根节点的kind值如何变化呢？

现有x和y，d为x和y的关系，xf和yf分别是x和y的根节点，于是我们有x--rank[x]-->xf，y--rank[y]-->yf，显然我们可以得到xf--(3-rank[x])-->x，yf--(3-rank[y])-->y。假如合并后x为新的树的根节点，那么原先fx树上的节点不需变化，yf树则需改变了，因为rank值为该节点和树根的关系。这里只改变rank(yf)即可，因为在进行find_set操作时可相应改变yf树的所有节点的kind值。于是问题变成了yf--?-->xf。我们不难发现yf--(3-rank[y])-->y--(3-d)-->x--rank[x]-->xf,根据前面的结论，我们有yf--(3-rank[y])-->y--(3-d)-->x--rank[x]-->xf。我们求解了xf和yf的关系了。

代码如下：

#include <iostream>
const int MAX=50005;

int father[MAX];
int rank[MAX];

//初始化集合
void Make_Sent(int x)
{
father[x]=x;
rank[x]=0;
}

//查找x的集合，回溯时压缩路径，并修改x与father[x]的关系
int Find_set(int x)
{
int t;
if(x!=father[x])
{
t = father[x];
father[x]= Find_set(father[x]);
//更新x与father[X]的关系
rank[x] = (rank[x]+rank[t])%3;
}
return father[x];
}

//合并x,y所在的集合
void Union(int x,int y,int d)
{
int xf = Find_set(x);
int yf = Find_set(y);
//将集合xf合并到yf集合上
father[xf] = yf;
//更新 xf 与father[xf]的关系
rank[xf]=(rank[y]-rank[x]+3+d)%3;
}

int main()
{
int totle=0;
int i,n,k,x,y,d,xf,yf;
scanf("%d%d",&n,&k);
for(i=1;i<=n;++i)
Make_Sent(i);
while(k--)
{
scanf("%d%d%d",&d,&x,&y);
//如果x或y比n大，或x吃x，是假话
if(x>n||y>n||(d==2 && x == y))
{
totle++;
}
else
{
xf = Find_set(x);
yf = Find_set(y);
//如果x，f的父节点相同 ，那么可以判断给出的关系是否正确的
if(xf == yf)
{
if((rank[x]-rank[y]+3)%3 != d-1)
totle++;
}
else
{
//否则合并x，y
Union(x,y,d-1);
}
}
}
printf("%d\n",totle);
system("pause");
return 0;
}

http://cavenkaka.iteye.com/blog/1489588