小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]);
}
}