百度之星 数据分割 并查集+set

小w来到百度之星的赛场上，准备开始实现一个程序自动分析系统。

这个程序接受一些形如 xi=xjxi=xj 或 xi≠xjxi≠xj 的相等/不等约束条件作为输入，判定是否可以通过给每个 w 赋适当的值，来满足这些条件。

输入包含多组数据。
然而粗心的小w不幸地把每组数据之间的分隔符删掉了。
他只知道每组数据都是不可满足的，且若把每组数据的最后一个约束条件去掉，则该组数据是可满足的。

请帮助他恢复这些分隔符。 Input第 11行：一个数字 LL，表示后面输入的总行数。

之后 LL行，每行包含三个整数， i,j,ei,j,e，描述一个相等/不等的约束条件，若 e=1e=1，则该约束条件为 xi=xjxi=xj ，若 e=0e=0，则该约束条件为 xi≠xjxi≠xj 。

i,j,L≤100000i,j,L≤100000

xi,xj≤Lxi,xj≤LOutput输出共 T+1T+1行。

第一行一个整数 TT，表示数据组数。

接下来 TT行的第 ii行，一个整数，表示第i组数据中的约束条件个数。
Sample Input

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

Sample Output

1
6

题解：因为要考虑新的关系是不是和之前的矛盾，但是如何记录不等的关系就很困难，这里使用set[x]记录与以x为根的并查集中的所有元素都不想等的元素。所以只需要在合并的时候，合并set即可。

#include <iostream>
#include<cstdio>
#include<set>
#include<algorithm>
using namespace std;
set<int>s[100005],se;
int p[100005];
int finde(int x){
	if(p[x]!=0)return p[x]=finde(p[x]);
	return x;
}
int join(int x,int y){
	if(x>y)swap(x,y);
	if(finde(x)==finde(y))return 0;
	set<int>::iterator it;
	for(it=s[x].begin();it!=s[x].end();it++){
		s[*it].erase(x);
		s[*it].insert(y);
		s[y].insert(*it);
	}
	p[x]=y;
	return 1;
}
int main()
{
	int l,i,ans[100005],cnt=0,a,b,typ,num=0;
	scanf("%d",&l);
	set<int>::iterator it;
	while(l--){
		cnt++;
		scanf("%d%d%d",&a,&b,&typ);
		se.insert(a),se.insert(b);
		a=finde(a);b=finde(b);
		if(typ){//相等
			if(a==b)continue;
			if(s[a].count(b)==0)join(a,b);
			else {
				ans[++num]=cnt;
				cnt=0;
				for(it=se.begin();it!=se.end();it++){
					s[*it].clear();
					p[*it]=0;
				}
				se.clear();
			}
		}
		else {
			if(a==b){
				ans[++num]=cnt;
				cnt=0;
				for(it=se.begin();it!=se.end();it++){
					s[*it].clear();
					p[*it]=0;
				}
				se.clear();
			}
			else {
				s[a].insert(b);
				s[b].insert(a);
			}
		}
	}
	printf("%d\n",num);
	for(i=1;i<=num;i++)printf("%d\n",ans[i]);
    return 0;
}