hdu 6109 数据分割(并查集+set合并)

思路：

如果是相等，则看作一个集合，如果不等，则将两个集合连边。

在判断的时候，如果相等的集合中操作是0，就不对；

如果不等的集合操作是1，就不对

#include <algorithm>
#include <cstring>
#include <set>
using namespace std;
const int maxn=100005;
set<int> g[maxn];
int n,m;
int a[maxn],b[maxn],c[maxn],fa[maxn];
int anst,ans[maxn];
int find(int x)
{
    if(fa[x]!=x)
        fa[x]=find(fa[x]);
    return fa[x];
}

void init()
{
    for(int i=1;i<maxn;i++)
        g[i].clear(),fa[i]=i;

} 

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d%d%d",&a[i],&b[i],&c[i]);
    init();
    set<int>::iterator it;
    for(int i=1;i<=n;i++)
    {
        int u=find(a[i]);
        int v=find(b[i]);
        if(c[i])
        {
            if(u==v)
                continue;
            else if(g[u].find(v)!=g[u].end())
            {
                init();
                ans[++anst]=i; 
            }
            else
            {
                for(it=g[v].begin();it!=g[v].end();++it)
                {
                    g[u].insert(*it);
                    g[*it].erase(v);
                    g[*it].insert(u);
                }
                g[v].clear();
                fa[v]=u;
            }
        }
        else
        {
            if(u!=v)
            {
                g[u].insert(v);
                g[v].insert(u);
            }
            else
            {
                ans[++anst]=i;
                init();
            }
        }
    }
    printf("%d\n",anst);
    for(int i=1;i<=anst;i++)
        printf("%d\n",ans[i]-ans[i-1]);
    return 0;
}