HDU 2473 Junk-Mail Filter(并查集设辅助点)

题目链接

题目大意

  有n个编号为0~n的物品，并且有两个操作，分别是将两个物品合并在一个集合里和把一个物品从集合里取出来，问最后有多少个集合。

解题思路

  本题主要是如何把物品从集合中取出，可以设立一个与原物品编号不同的点作为辅助点。这样挂在同一个辅助点的物品都是属于同一个集合的，而挂在不同辅助点的物品属于不同的集合。

代码

const int maxn = 2e6+10;
int n, m, kase=1, p[maxn];
int find(int x) {
    return p[x]==x ? p[x]:p[x]=find(p[x]);
}
int main(void) {
    while(~scanf("%d%d",&n,&m)&&(n||m)) {
        char ch[2];
        for (int i = 0; i<n; ++i) {
            p[i] = i+n; p[i+n] = i+n;
        }
        int tot = 2*n;
        for (int i = 0,a,b; i<m; ++i) {
            scanf("%s",ch);
            if(ch[0]=='M') {
                scanf("%d%d",&a,&b);
                p[find(a)] = find(b);
            }
            else {
                scanf("%d",&a);
                p[tot] = tot;
                p[a] = tot++;
            }
        }
        set<int> cnt;
        for (int i = 0; i<n; ++i)
            cnt.insert(find(p[i]));
        int ans = cnt.size();
        cnt.clear();
        printf("Case #%d: %d
", kase++,ans);
    }
    return 0;
}