hdu 2473 Junk-Mail Filter

http://acm.hdu.edu.cn/showproblem.php?pid=2473

并查集设置虚拟父节点。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 1100000
using namespace std;

int f[maxn],num[maxn];
int in[maxn],n,m;

void inti()
{
    for(int i=0; i<n; i++)
    {
        f[i]=in[i]=i;
        num[i]=1;
    }
}

int find1(int x)
{
    if(x==f[x]) return x;
    return f[x]=find1(f[x]);
}

void merge1(int a,int b)
{
    int fa=find1(a);
    int fb=find1(b);
    if(fa!=fb)
    {
        f[fa]=fb;
        num[fb]+=num[fa];
        num[fa]=0;
    }
}

int main()
{
    int ca=1;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        inti();
        if(n==0&&m==0) break;
        getchar();
        char ch;
        int ans=n;
        int u,v;
        for(int i=1; i<=m; i++)
        {
            scanf("%c",&ch);
            if(ch=='M')
            {

                scanf("%d%d",&u,&v);
                merge1(in[u],in[v]);
            }
            else if(ch=='S')
            {
                scanf("%d",&v);
                int f1=find1(in[v]);
                num[f1]--;
                in[v]=ans;
                f[ans]=ans;
                num[ans]=1;
                ans++;
            }
            getchar();
        }
        int t1=0;
        for(int i=0; i<ans; i++)
        {
            if(num[i]>0) t1++;
        }
        printf("Case #%d: %d\n",ca++,t1);
    }
    return 0;
}