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

题目链接：hdu 6109 数据分割

题意：

给你一些不等和相等关系，然后分割，每次都按第一个不成立的等式分割。

题解：

由于相等关系具有传递性，不等关系不具有传递性。

所以相等关系可以由并查集来存储，不相等关系用set来存，目的是为了让查找log(n)

然后在两个集合进行合并的时候，要采用启发式合并的方式，及每次都将小的集合的每个元素放进大的集合。

这样启发式合并的总复杂度为nlog(n)。然后加上set的insert复杂度，总负责度就是n*log(n)²

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;

const int N=1e5+7;
set<int>st[N];
int n,ans[N],ed,cnt,a,b,c,f[N],S[N],top;

int find(int x){return f[x]==x?x:f[x]=find(f[x]);}
void get(){
    ans[++ed]=cnt,cnt=0;
    while(top)
    {
        f[S[top]]=S[top];
        st[S[top]].clear();
        top--;
    }
}
void merge(int x,int y)
{
    if(st[x].size()<st[y].size())swap(x,y);
    for(int it:st[y])st[x].insert(find(it));
    f[y]=x;
}

int main(){
    while(~scanf("%d",&n))
    {
        F(i,1,n)f[i]=i,st[i].clear();
        ed=0,cnt=0,top=0;
        F(i,1,n)
        {
            scanf("%d%d%d",&a,&b,&c);
            cnt++,S[++top]=a,S[++top]=b;
            if(a==b&&c==1)continue;
            if(a==b&&c==0){get();continue;}
            if(c==1)
            {
                int fx=find(a),fy=find(b);
                if(st[fx].find(fy)!=st[fx].end())get();
                else merge(fx,fy);
            }else
            {
                int fx=find(a),fy=find(b);
                if(fx==fy)get();
                else st[fx].insert(fy),st[fy].insert(fx);
            }
        }
        printf("%d\n",ed);
        F(i,1,ed)printf("%d\n",ans[i]);
    }
    return 0;
}