hdu 6109 数据分割（并查集+set合并）

数据分割

Problem Description

小w来到百度之星的赛场上，准备开始实现一个程序自动分析系统。

这个程序接受一些形如xi=xj 或 xi≠xj 的相等/不等约束条件作为输入，判定是否可以通过给每个 w 赋适当的值，来满足这些条件。

输入包含多组数据。
然而粗心的小w不幸地把每组数据之间的分隔符删掉了。
他只知道每组数据都是不可满足的，且若把每组数据的最后一个约束条件去掉，则该组数据是可满足的。

请帮助他恢复这些分隔符。

Input

第1行：一个数字L，表示后面输入的总行数。

之后L行，每行包含三个整数，i,j,e，描述一个相等/不等的约束条件，若e=1，则该约束条件为xi=xj ，若e=0，则该约束条件为 xi≠xj 。

i,j,L≤100000

xi,xj≤L

Output

输出共T+1行。

第一行一个整数T，表示数据组数。

接下来T行的第i行，一个整数，表示第i组数据中的约束条件个数。

Sample Input

6
2 2 1
2 2 1
1 1 1
3 1 1
1 3 1
1 3 0

Sample Output

1
6

思路：相等关系是可传递性的，因此我们可以用并查集来维护。而不等关系是不可传递性的，我们可以用set来维护。

这样，依次遍历所有条件，每次遇到一个与已有条件冲突的条件时，就将其进行分割，同时初始化所有值。

注意：如果依次有条件$x1!=x3,x2!=x4,x1=x2$，当我们将$x1$和$x2$合并的时候（比如将$x1$合并到$x2$上），我们也要将$x1$的不等关系合并到$x2$上

代码：

#include<bits/stdc++.h>
using namespace std;

const int maxn=1e5+10;
set<int>s[maxn],num;
int pre[maxn],ans[maxn];
int tot,cnt;

void init()
{
    for(int i=1; i<maxn; ++i)
        pre[i]=i;
}

int Find(int x)
{
    if(x!=pre[x])
        pre[x]=Find(pre[x]);
    return pre[x];
}

void Init()//对修改过的元素初始化
{
    for(set<int>::iterator iter=num.begin(); iter!=num.end(); ++iter)
    {
        pre[*iter]=*iter;
        s[*iter].clear();
    }
    num.clear();
}

int main()
{
    init();
    int t,x,y,w;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&x,&y,&w);
        ++cnt;
        num.insert(x),num.insert(y);
        int fx=Find(x),fy=Find(y);
        if(w)
        {
            if(s[fx].count(fy)||s[fy].count(fx))
            {
                ans[tot++]=cnt;
                cnt=0;
                Init();
                continue;
            }
            if(fx!=fy)
            {
                if(s[fx].size()>s[fy].size())
                    swap(fx,fy);
                pre[fx]=fy;
                for(set<int>::iterator iter=s[fx].begin(); iter!=s[fx].end(); ++iter)
                {
                    s[*iter].erase(fx);
                    s[*iter].insert(fy);
                    s[fy].insert(*iter);
                }
                s[fx].clear();
            }
        }
        else
        {
            if(fx==fy)
            {
                ans[tot++]=cnt;
                cnt=0;
                Init();
                continue;
            }
            s[x].insert(y),s[y].insert(x);
        }
    }
    printf("%d\n",tot);
    for(int i=0; i<tot; ++i)
        printf("%d\n",ans[i]);
    return 0;
}