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;}