【ZCMU1435】盟国（并查集删除节点）

题目链接

1435: 盟国

Time Limit: 3 Sec  Memory Limit: 128 MB
Submit: 456  Solved: 104
[Submit][Status][Web Board]

Description

世界上存在着N个国家，简单起见，编号从0~N-1，假如a国和b国是盟国，b国和c国是盟国，那么a国和c国也是盟国。另外每个国家都有权宣布退盟（注意，退盟后还可以再结盟）。

定义下面两个操作：

“M X Y” ：X国和Y国结盟  (如果X与Z结盟,Y与Z结盟,那么X与Y也自动结盟).

“S X” ：X国宣布退盟 (如果X与Z结盟,Y与Z结盟,Z退盟,那么X与Y还是联盟).

 

Input

多组case。

每组case输入一个N和M (1 ≤ N ≤ 100000 , 1 ≤ M ≤ 1000000),N是国家数，M是操作数。

接下来输入M行操作

当N=0，M=0时，结束输入

 

Output

对每组case输出最终有多少个联盟(如果一个国家不与任何国家联盟,它也算一个独立的联盟)，格式见样例。

 

Sample Input

5 6

M 0 1

M 1 2

M 1 3

S 1

M 1 2

S 3

3 1

M 1 2

0 0

Sample Output

Case #1: 3 Case #2: 2

HINT

 

 带删除并查集

 

Source

FZU

 

【解题思路】

pre数组代表每个节点的根节点，a数组为节点编号，当某个节点被删除后，因为其他关系保持不变，所以只需将被删除节点的编号就记为比n大的数，方便之后的统计。

 

【代码】

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e6+5;
int pre[maxn],a[maxn];
 
int findroot(int x){
    return pre[x]==x?pre[x]:pre[x]=findroot(pre[x]);
}
int main()
{
    int n,m,kase=1;
    while(~scanf("%d%d",&n,&m) && n || m)
    {
        for(int i=0;i<maxn;i++)
            pre[i]=i;
        for(int i=0;i<n;i++)
            a[i]=i;
        int num=n;
        while(m--)
        {
            char str[2];
            scanf("%s",str);
            if(str[0]=='M')
            {
                int u,v;
                scanf("%d%d",&u,&v);
                int fu=findroot(a[u]);
                int fv=findroot(a[v]);
                if(fu!=fv)pre[fu]=fv;
            }
            else
            {
                int x;
                scanf("%d",&x);
                a[x]=num++;
            }
        }
        set<int>s;
        for(int i=0;i<n;i++)
            s.insert(findroot(a[i]));
        printf("Case #%d: %d\n",kase++,s.size());
    }
    return 0;
}