C - 数据分割

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

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

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

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

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

i,j,L≤100000

xi,xj≤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和并查集都重置了

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5; 
set<int> s[N];
set<int> s1;
int x,y,e,l;
int ans[N];
int f[N];
int find(int x)
{
	return x==f[x]?x:f[x]=find(f[x]);
}
int main()
{
	while(~scanf("%d",&l))
	{
		int num=0;
		int cnt=0;
		for(int i=1;i<=N;i++)
		f[i]=i;
		while(l--)
		{
			cnt++;
			int flag=1;
			scanf("%d%d%d",&x,&y,&e);
			{
				s1.insert(x);
				s1.insert(y);
				if(e==0)
				{
					int fx=find(x);
					int fy=find(y);
					if(fx==fy)
					flag=0;
					else
					{
						s[fx].insert(y);
						s[fy].insert(x);
					}
				}
				else
				{
					int fx=find(x);
					int fy=find(y);
					if((s[fy].find(x)!=s[fy].end())||(s[fx].find(y)!=s[fx].end()))
					flag=0;
					else
					{
						if(s[fx].size()>s[fy].size())
						{
							s[fx].insert(s[fy].begin(),s[fy].end());
							f[fy]=fx;
						}
						else
						{
							s[fy].insert(s[fx].begin(),s[fx].end());
							f[fx]=fy;
						}
					}
				}
			}
			if(flag)
			continue;
			ans[++num]=cnt;
			cnt=0;
			for(set<int>::iterator it=s1.begin();it!=s1.end();it++) 
			{
                int tmp=*it;
                f[tmp]=tmp;
                s[tmp].clear();
            }
            s1.clear();
		}
		printf("%d\n",num);
		for(int i=1;i<=num;i++)
		printf("%d\n",ans[i]);		
	}
}