hdu6109 数据分割【并查集+set】

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6109
题意：中文题
解析：由于相等具有传递性，那么可以选择用并查集来维护，即若干个相等的就是一堆，那么对于不等号来说，是无法传递的，于是可以选择用set来维护，比如对于x这个集合来说，里面的所有元素都是与x不等的。这些是对于数据结构的选择，那么有一些要注意的点就是，当你两个相等的元素合并时，其不等的元素也要往父节点合并，不如，x1=y1，x1≠x2，y1≠y2，那么也就有x1≠y2，y1≠x2
主要就是这些，接着边输入边处理即可

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+100;
int fa[maxn];
set<int>s[maxn];
vector<int>ans;
void init(int n)
{
    for(int i=0;i<=n;i++)
    {
        fa[i] = i;
        s[i].clear();
    }
}
void push_up(int x,int fa)
{
    set<int>::iterator it = s[x].begin();
    for(;it!=s[x].end();it++)
        s[fa].insert(*it);
}
int getfa(int x)
{
    if(x==fa[x])
        return fa[x];
    push_up(x,fa[x]);
    return fa[x] = getfa(fa[x]);
}
void merge(int u,int v)
{
    int t1 = getfa(u);
    int t2 = getfa(v);
    if(t1!=t2)
        fa[t1] = t2;
}
int main(void)
{
    int n;
    scanf("%d",&n);
    init(n);
    int cnt = 0;
    for(int i=0;i<n;i++)
    {
        int x,y,e;
        scanf("%d %d %d",&x,&y,&e);
        cnt++;
        x = getfa(x);
        y = getfa(y);
        if(e)
        {
            if(x==y)
                continue;
            if(s[x].find(y)!=s[x].end() || s[y].find(x)!=s[y].end())
            {
                ans.push_back(cnt);
                cnt = 0;
                init(n);
            }
            else
                merge(x,y);
        }
        else
        {
            if(x==y)
            {
                ans.push_back(cnt);
                cnt = 0;
                init(n);
            }
            else
            {
                s[x].insert(y);
                s[y].insert(x);
            }
        }
    }
    printf("%d\n",ans.size());
    for(int i=0;i<(int)ans.size();i++)
        printf("%d\n",ans[i]);
    return 0;
}