SDNUOJ 1076.种类判断(种类并查集)

Time Limit: 1000 MS Memory Limit: 32768 KB

Description
已知有 n 个学生，其中有男生也有女生，给定 m 个关系，每个关系表示两个同学性别是否相同。如 同学A 和同学 B 是同性，或不是同性，判断这些条件中是否存在矛盾。

Input
第一行为两个整数 n(1 <= n <= 1000)， m(1 <= m <= 1000)，每个同学编号为1到n，之后m行，每行三个整数，前两个整数表示两个同学的编号，第三个整数为0或者1，若为0，则表示两个同学为同性，否则为异性。

Output
一行，若存在矛盾，输出YES，否则输出NO

Sample Input
5 5
1 2 1
2 3 1
3 4 1
4 5 1
5 1 0

Sample Output
NO

#include<iostream>
using namespace std;
int p[2005];//开数据量两倍大的数组（种类并查集） 
void init(int n)    //初始化 
{
	for(int i=1;i<=n;++i)
	{
		p[i]=i;
	}
}
int find(int v)
{
	if(p[v]==v) return v;
	else
	{
		p[v]=find(p[v]);   //路径压缩 
		return p[v];
	}
}
void merge(int x,int y)
{
	int t1=find(x),t2=find(y);
	if(t1!=t2)
	{
		p[t2]=t1;   //合并操作合并的是俩根节点（一定要注意） 
	}
}
int main()
{
	int n,m,a,b,c;
	while(cin>>n>>m)
	{
	bool flag=0;
	init(n*2);         //k+n与k种类不同 
	for(int i=1;i<=m;++i)
	{
		cin>>a>>b>>c;
		if(a<=0||a>n||b<=0||b>n)  //判断数据是否越界 
		{
			flag=1;
		}
		if(c==1)
		{
			if(find(a)==find(b)||find(a+n)==find(b+n))  //这两种情况都代表a和b是同类 
			{
				flag=1;
			}
			merge(a,b+n),merge(a+n,b);   //种类并查集的合并操作 ，前面的好理解，后面的就是敌人的朋友的敌人和自己是同类= = 
		}
		else
		{
			if(find(a)==find(b+n)||find(b)==find(a+n))//这两种情况都代表a，b是异类，因为敌人的同类是敌人 
			{
				flag=1;
			}
			merge(a,b),merge(a+n,b+n);   //种类并查集的操作 
		}
	}
	if(flag) cout<<"YES\n";
	else cout<<"NO\n";
	}
	return 0;
}