【百度之星初赛A】1002 数据分割(并查集+Set)

在比赛的时候直接想到的就是用并查集加Set的做法，但是由于评测爆炸，只交了一发还没有过。赛后有好好想了之前的思路，还参考了一些博客，发现这种做法的确可以AC。

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

分析：

如果两个元素是相等的那就将它们合并到一起，同时用Set记录不相等的数，开一个Set数组，Set[i]表示与i不相等的数。因此相连的都是相等的，集合中的都是不等的。

如果e = 1, 那就找a, b的根节点fa, fb，如果在Set[fa]中找到fb 或者 在Set[fb]中找到fa， 那就说明存在矛盾，就从这里截断。如果没有矛盾，那就判断两个是不是同根，如果不同根就合并两个集合，并且更新相关的Set[i]

如果e = 0, 先把fa加到Set[fb]中， fb加到Set[fa]中，然后判断两个是不是同根，如果同根就有矛盾。

就这样就可以完成数据分割。

同时，还看到了一个初始化的技巧，就是将所有出现的数字，用Set记录，当出现矛盾，要开始下一组时只需要将Set中的每一个元素对应的Set和数组置位就可以了。

代码：

#include <iostream>#include <cstdio>#include <cstdlib>#include <algorithm>#include <cstring>#include <map>#include <set>#include <vector>#include <queue>#include <cmath>#include <bitset>using namespace std;const int MAXN = 100005;int fa[MAXN];set<int> Set[MAXN], con;int re[MAXN];int find(int x){    if(fa[x] == -1) return x;    return fa[x] = find(fa[x]);}void unit(int x, int y){    x = find(x);    y = find(y);    if(x != y) fa[x] = y;}int main(){    int L;    re[0] = 0;    int cnt = 0;    scanf("%d", &L);    int i, j, e;    memset(fa, -1, sizeof(fa));    while(L--)    {        scanf("%d%d%d", &i, &j, &e);        con.insert(i), con.insert(j);        int ri = find(i), rj = find(j);        cnt += 1;        if(e == 1)        {            if(Set[ri].count(rj) || Set[rj].count(ri))            {                re[++re[0]] = cnt;                cnt = 0;                for(auto e : con)                {                    Set[e].clear();                    fa[e] = -1;                }                con.clear();            }            else if(ri != rj)            {                fa[ri] = rj;                for(auto e : Set[ri])                {                    Set[rj].insert(e);                    Set[e].erase(ri);                    Set[e].insert(rj);                }                Set[ri].clear();            }        }        else        {            Set[ri].insert(rj), Set[rj].insert(ri);            if(ri == rj)            {                re[++re[0]] = cnt;                cnt = 0;                for(auto e : con)                {                    Set[e].clear();                    fa[e] = -1;                }                con.clear();            }        }    }    for(int i = 0; i <= re[0]; ++i)        printf("%d\n", re[i]);    return 0;}