[HDU 6109]数据分割（启发式合并）

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

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

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

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

之后L行，每行包含三个整数，$i,j,e$，描述一个相等/不等的约束条件，若$e=1$，则该约束条件为$x_i=x_j$ ，若$e=0$，则该约束条件为 $x_i\ne x_j$ 。

$i,j,L\le 100000$

$x_i,x_j\le L$

Output
输出共T+1行。

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

接下来T行的第$i$行，一个整数，表示第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来记录不等关系，由于相等关系是传递的，每次得到两个元素相等，则检查它们所在的并查集树根对应的set，看是否冲突，不冲突则合并这两棵树，同时合并set，冲突则代表找到了一组数据。每次得到两个元素不等，则检查它们是否在同一个并查集，若在则代表找到了一组数据，否则分别将冲突关系记入树根所对应的set。

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
set<int>s[100005];
int f[100005];
int que[200005],a,b,c,cot;
int an[100005],ans_cot;
int findf(int now)
{
	if(f[now]==now)return now;else return f[now]=findf(f[now]);
}
void clear()
{
	for(int i=1;i<=cot;i++) 
	{
		set<int>().swap(s[que[i]]);
		f[que[i]]=que[i];
	}
	cot=0;
}
int judge_equal(int l,int r)
{
	if(findf(l)==findf(r))return 0;
	if(s[findf(l)].size()<s[findf(r)].size()) swap(l,r);
	for(set<int>::iterator it=s[findf(r)].begin();it!=s[findf(r)].end();it++)
	{
		if(findf(*it)==findf(l))return 1;
	}
	for(set<int>::iterator it=s[findf(r)].begin();it!=s[findf(r)].end();it++)
	{
		s[findf(l)].insert(*it);
	}
	set<int>().swap(s[findf(r)]);
	f[findf(r)]=f[findf(l)];
	return 0;
}
int judge_unequal(int l,int r)
{
	if(findf(l)==findf(r))return 1;
	s[findf(l)].insert(r);
	s[findf(r)].insert(l);
	return 0;
}
int main()
{
	int l;
	for(int i=1;i<=100000;i++) f[i]=i;
	scanf("%d",&l);
	for(int i=1;i<=l;i++)
	{
		scanf("%d%d%d",&a,&b,&c);
		que[++cot]=a;
		que[++cot]=b;
		int temp;
		if(c) temp=judge_equal(a,b);else temp=judge_unequal(a,b);
		if(temp)
		{
			an[ans_cot++]=cot/2;clear();
		}
	}
	printf("%d\n",ans_cot);
	for(int i=0;i<ans_cot;i++)printf("%d\n",an[i]);
	return 0;
}