hdu 6109
思路:尽量让多的条件满足,当有一个条件和前面的冲突时,就是分隔符的时候,用并查集维护相同的数,set维护不同的数。当两个数在同一个并查集中时,代表两个数相同,不在同一个并查集是说明两个数可能相同也可能不同。当a在b的set中时代表b和a不相同。所以当我们每次判断两个数是否相同时先找并查集,并且要顺带更新set因为a==b,
b≠c
,所以
a≠c
但是这样每次更新一个并查集里面的所有元素复杂度太高一不好写,所以当每次找的时候我们只需要将子节点的set加到父节点的set理解好了然后判断两个祖先节点的set是否有出现过,有的话就代表
a≠b
。
#include<set>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
using namespace std;
const int maxn=1e6+10;
int fa[maxn],dep[maxn];//dep记录的是并查集的高度,用于将低树并到高树上
set<int> s[maxn/10];//s[i]记录和i不相等的集合
void init(int n)//初始化
{
for(int i=1;i<=n;i++)
{
fa[i]=i;
dep[i]=0;
s[i].clear();
}
}
void push_up(int a,int b)
{
for(set<int>::iterator it=s[a].begin();it!=s[a].end();it++)
s[b].insert(*it);
}
int finds(int x)
{
if(x==fa[x])
return x;
push_up(x,fa[x]);//将儿子不相等的并到父亲的集合里
return fa[x]=finds(fa[x]);
}
void merger(int a,int b)//合并a,b
{
a=finds(a),b=finds(b);
if(dep[a]<dep[b])
fa[a]=b;
else
{
fa[b]=a;
if(dep[a]==dep[b])
dep[a]++;
}
}
int ans[maxn];
int main()
{
int n;
while(~scanf("%d",&n))
{
init(n);
int cont=0,tot=0;
for(int i=0;i<n;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
a=finds(a),b=finds(b);
tot++;
if(c)
{
if(a!=b)//当fa[a]和fa[b]不相同时a可能等于b也可能不等于
{
if(s[a].count(b)||s[b].count(a))//fa[a]中有fa[b]或者fa[b]中有fa[a]则说明a和b不相等
{
ans[cont++]=tot;
tot=0;
init(n);
}
else//a和b相同
merger(a,b);
}
}
else
{
if(a==b)//fa[a]和fa[b]在同一个并查集里说明a==b
{
ans[cont++]=tot;
tot=0;
init(n);
}
else
{
s[a].insert(b);
s[b].insert(a);
}
}
}
printf("%d\n",cont);
for(int i=0;i<cont;i++)
printf("%d\n",ans[i]);
}
}