B:Butterfly

B:Butterfly

总时间限制: 

1000ms

单个测试点时间限制: 

100ms

内存限制: 

65536kB

描述

有一群旅行爱好者，有一天，他们带回了n只蝴蝶回来。他们相信每一只都属于两个不同种类中的一种，为了讨论方便，我们称它们为A与B。他们想把n只标本分成两组——一些属于A且一些属于B——但是直接标记任何一个标本对于他们是非常困难，因此他们决定采用下面的方法。

对每对标本i和j，他们细心地把它们放到一起研究。如果他们以自己的判断足以确信，那么他们把这对蝴蝶标记为“相同”（这意味着他们相信这两只来自同一类）或者是“不同”（这意味着他们相信这两只来自不同的类）。他们也可以对某些标本判断不出来而弃权，在这种情况下，我们称这对标本是不明确的。

现在他们有n只标本的集合，还有对那些没有弃权的标本对的m个判断的集合（“相同”或者“不同”）。他们想知道这个数据与每只蝴蝶来自A和B中的一个类的想法是否一致。更具体地说，如果对每对蝴蝶按照下述方式标记A或B是可能的，即对每个标为“相同”的（i，j）对，就是i与j有相同标记的情况；对每个标为“不同”的（i，j）对，就是i与j有不同标记的情况。那么我们可以说这m个判断是一致的。他们正在冥思苦想自己的判断是否是一致的。请你帮他们设计合理的算法解决该问题。
 

输入

输入包含多组数据，以文件结束符为终止。

每组数据第一行为两个整数，分别是n和m：
n为蝴蝶的数量，编号从0到n-1
m为关系的数量

接下来是m组关系数据，每组数据占一行，为三个整数，前两个整数表示蝴蝶的编号，第三个整数为关系的种类（相同或者不同）：
0为相同，1为不同

1 < n <= 1000
1 < m <= 100000

输出

合理就输出YES
不合理就输出NO

样例输入

3 3
0 1 0
1 2 1
0 2 1
3 3
0 1 0
1 2 1
0 2 0

样例输出

YES
NO

提示

注意输出的大写和回车

题解：

这个题的数据很水，用并查集只要判断最后集合数量大于3输出NO也可以ac，但是个人以为这种方式不对，用并查集最后还是要染色，不如直接用BFS染色

#include<bits/stdc++.h>

using namespace std;

int main()
{
	//freopen("in.txt","r",stdin);
	int m,n;
	while(scanf("%d %d",&n,&m)!=EOF)
	{
		bool f=true;
		queue<int> q;
		int color[n]={0};
		int a[n][n]={0};
		for(int i=0;i<m;i++)
		{
			int aa,b,c;
			scanf("%d %d %d",&aa,&b,&c);
			if(c==0) 
				c=1;
			else 
				c=2;
			a[aa][b]=c;
			a[b][aa]=c;
		}
		for(int k=0;k<n;k++)
		{
			if(color[k])
				continue;
			color[k]=1;
			q.push(k);
			while(!q.empty())
			{
				int t=q.front();
				q.pop();
				for(int j=0;j<n;j++)
				{
					if(a[t][j]==1)// same set
					{
						if(color[j]==0-color[t])
						{
							cout<<"NO"<<endl;
							f=false;
							break;
						}
						else if(color[j]==0)
						{
							color[j]=color[t];
							q.push(j);
						}
					}
					else if(a[t][j]==2)//diff set
					{
						if(color[j]==color[t])
						{
							cout<<"NO"<<endl;
							f=false;
							break;
						}
						else if(color[j]==0)
						{
							color[j]=0-color[t];
							q.push(j);
						}
					}
				}
				if(!f)
					break;
			}
			if(!f)
				break;
		}
		if(f)
			cout<<"YES"<<endl;
	}
}