POJ 1182 食物链（并查集）

动物王国中有三类动物A,B,C，这三类动物的食物链构成了有趣的环形。A吃B， B吃C，C吃A。
现有N个动物，以1－N编号。每个动物都是A,B,C中的一种，但是我们并不知道它到底是哪一种。
有人用两种说法对这N个动物所构成的食物链关系进行描述：
第一种说法是"1 X Y"，表示X和Y是同类。
第二种说法是"2 X Y"，表示X吃Y。
此人对N个动物，用上述两种说法，一句接一句地说出K句话，这K句话有的是真的，有的是假的。当一句话满足下列三条之一时，这句话就是假话，否则就是真话。
1） 当前的话与前面的某些真的话冲突，就是假话；
2） 当前的话中X或Y比N大，就是假话；
3） 当前的话表示X吃X，就是假话。
你的任务是根据给定的N（1 <= N <= 50,000）和K句话（0 <= K <= 100,000），输出假话的总数。

Input

第一行是两个整数N和K，以一个空格分隔。
以下K行每行是三个正整数 D，X，Y，两数之间用一个空格隔开，其中D表示说法的种类。
若D=1，则表示X和Y是同类。
若D=2，则表示X吃Y。

Output

只有一个整数，表示假话的数目。

Sample Input

100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

Sample Output

3

上个周末有幸参加了cccc和电子科大的一个比赛，感觉自己拖了队伍的后腿，挺不甘心的。自己还有很长的路要走啊。加油加油，争取下次能够拿个奖回来。

不抱怨了，写正文。

先讲讲什么是并查集。并查集是一个用于管理元素分组情况的数据结构，有两个高效的操作：

1、查询元素a和b是否属于同一组。

2、合并a和b所在的组。

每组数据用树的结构来储存，查询功能就是找这两个元素是否有相同的根。

合并就是那一组数的根连向另一组树的根，合并成一棵树。

下面是并查集的实现：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int par[1000000];
int rank[1000000];
//初始化
void init(int n)
{
 int i;
 for(i=0;i<n;i++)
 {
  par[i]=i;
  rank[i]=0;
 }
}
//查询树跟
int find(int x)
{
 if(par[x]==x){return x;}
 else
 {
  return par[x]=find(par[x]);
 }
}
//合并x,y
void unite(int x,int y)
{
 x=find(x);
 y=find(y);
 if(x==y){return ;}
 if(rank[x] < rank[y])
 {
  par[x]=y;
 }
 else
 {
  par[y]=x;
  if(rank[x]==rank[y]){rank[x]++;}
 }
}
//判断x,y是否在同一组
bool same(int x,int y)
{
 return find(x)==find(y);
}

现在来讲讲上面的查询函数。假设有这样的一组数4 3 2 1，4的父亲是3，3的父亲是2，2的父亲是1，1是根。现在查4：

par[4]=3，执行第二个指令par[4]=find(par[4])也就是find(3)，par[3]=2，find(par[3])也就是 find(2)，pa[2]=1，find(1)

最后par[1]=1，也就是说1的父亲是它自己，那它就是根，返回1。所以4的根为1。

现在用并查集来做这题，就觉得很方便了。

思路：

首先，对于每只动物要创建3个元素i-A，i-B，i-C，表示它所属的种类。（这里可以初始化3*n个元素，i，i+n，i+2*n来表示所属种类）

对于第一种说法，我们就合并x-A和y-A， x-B和y-B，x-C和y-C。

对于第二种说法，我们就合并x-A和y-B，x-B和y-C，x-C和y-A。

再合并之前，还要先判断信息是否正确，例如在第一种信息中，如果x-A和y-B同一组，那就是错的。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int par[1000000];
int rank[1000000];
void init(int n)
{
int i;
for(i=1;i<=n;i++)//注意这里初始化从1开始,方便一一对应
{
par[i]=i;
rank[i]=0;
}
}
int find(int x)
{
if(par[x]==x){return x;}
else
{
return par[x]=find(par[x]);
}
}
void unite(int x,int y)
{
x=find(x);
y=find(y);
if(x==y){return ;}
if(rank[x] < rank[y])
{
par[x]=y;
}
else
{
par[y]=x;
if(rank[x]==rank[y]){rank[x]++;}
}
}
bool same(int x,int y)
{
return find(x)==find(y);
}
int main()
{
int n,k;
int x,y,t;
int ans=0;
scanf("%d %d",&n,&k);
init(3*n);
while(k--)
{
scanf("%d %d %d",&t,&x,&y);
if(x>n || y>n){ans++;continue;}
if(t==1)//第一种说法
{
if(same(x,y+n) || same(x,y+2*n))//如果是x吃y或y吃x的情况,错(那x和y绝对不会同类)
{
ans++;
}
else
{
unite(x,y);//合并正确的说法
unite(x+n,y+n);
unite(x+2*n,y+2*n);
}
}
else//第二种说法
{
if(same(x,y) || same(x,y+2*n))//如果是x和y同类或y吃x的情况,错(这里的y如果等于x,那也可以认为是x吃x的情况)
{
ans++;
}
else
{
unite(x,y+n);//合并正确的说法
unite(x+n,y+2*n);
unite(x+2*n,y);
}
}
}
printf("%d\n",ans);
return 0;
}

因为合并的时候是把x，x+n，x+2*n   3种情况都合并了，所以判断的时候只需要考虑一种情况就好。