hdu6109数据分割（并查集+set）

一道好题，忍不住转载记录下。

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6109

思路（转）：

既然题目说每组数据把最后一个约束条件去掉就合理

那么只要将这L组约数条件从前往后遍历一下就好了

初始化一个空集，将约束条件一个一个加入集合里，当加到某个约束条件之后不合理了，

就说明这是最后一个约束条件，把它删掉，之后集合里剩下的约束条件个数就是当前答案

然后清空集合，继续

主要是如何判断它突然就不合理了

很显然相等是有传递性的，也就是说这个并查集就能轻松搞定

但是不相等不具有传递性

那么考虑暴力，开100000个set，set[x]里面存的是所有与x不相等的数

那么有题解：

每个约束x≠y，只要看x和y是否在同一个并查集里就好，如果在同一并查集说明不合理

否则合理，set[x].insert(y), set[y].insert(x)

每个约束x==y，看set[x]里是否有y，set[y]里是否有x，如果有则不合理，否则将x和y加入并查集

但是这样是错的，上面紫色的部分错了！

因为有这种情况：x和y相等，x和z不相等，那么y和z肯定也不相等

也就是说只要z和x不相等，那么和x在同一个并查集合里的所有数都和z不相等

这个时候要将x并查集里所有的数都insert(z)，肯定超时

所以只能将所有的并查集归为一类，set[x]里面存的是所有与x并查集里面的数不相等的数

然后并查集合并的时候set跟着一起合并

因为最多只会有L对约束关系，出现最多2L个数，而并查集是O(n)的

所以处理的好的话复杂度是基本上线性的，只有额外set的复杂度

不要轻易memset

代码：

#include <iostream>
#include <cstdio>
#include <set>
#include <cstring>

using namespace std;
const int maxn = 100000+10;
int f[maxn],ans[maxn];
set<int> s[maxn];
int fi(int x)
{
    if(f[x]==-1)
        return x;
    return f[x] = fi(f[x]);
}

void uni(int x,int y)
{
    if(x==y)
        return;
    set<int>::iterator it;
    if(s[x].size()<s[y].size())
    {
        for(it=s[x].begin();it!=s[x].end();it++)
        {
            s[*it].erase(x);
            s[*it].insert(y);
            s[y].insert(*it);
        }
        f[x] = y;
    }
    else
    {
        for(it=s[y].begin();it!=s[y].end();it++)
        {
            s[*it].erase(y);
            s[*it].insert(x);
            s[x].insert(*it);
        }
        f[y] = x;
    }
}

int main()
{
    int l;
    scanf("%d",&l);
    memset(f,-1,sizeof(f));
    set<int> st;
    int sum = 0,cnt = 0;
    set<int>::iterator it;
    while(l--)
    {
        sum++;
        int a,b,e;
        scanf("%d%d%d",&a,&b,&e);
        st.insert(a),st.insert(b);
        int t1 = fi(a),t2 = fi(b);
        if(e==1)
        {
            if(t1==t2||s[t1].count(t2)==0)
            {
                uni(t1,t2);
            }
            else
            {
                ans[cnt++] = sum;
                sum = 0;
                for(it=st.begin();it!=st.end();it++)
                {
                    f[*it] = -1;
                    s[*it].clear();
                }
                st.clear();
            }
        }
        else
        {
            if(t1==t2)
            {
                ans[cnt++] = sum;
                sum = 0;
                for(it=st.begin();it!=st.end();it++)
                {
                    f[*it] = -1;
                    s[*it].clear();
                }
                st.clear();
            }
            else
            {
                s[t1].insert(t2);
                s[t2].insert(t1);
            }
        }
    }
    printf("%d\n",cnt);
    for(int i = 0;i<cnt;i++)
        printf("%d\n",ans[i]);
    return 0;
}