並查集 Junk-Mail Filter hdu 2473

輸入N M，表示N個點，M組數據

接下來M組數據

M表示連接A B

S表示刪除點A

計算共有多少組邊

提示

1）刪除點不是將與它相連的邊剪斷，而只是消除這個點

2）開兩個數組，f[maxn]和p[maxn]，表示i對應的真正值和i的父節點

#include <stdio.h>
#include <string.h>
#define maxn 1000001
int f[maxn];               //表示i對應的真實值
int p[maxn];               //表示i的父節點
int flag[maxn];            //計算多少組
int find(int i)           //查找根節點
{
	if(i==p[i])
		return p[i];
	p[i]=find(p[i]);
	return p[i];
}

int main()
{
	int n,k;
	int i,j;
	char s[3];
	int a,b;
	int fr,ed;
	int sum;
	int time=0;
	while(scanf("%d%d",&n,&k)!=-1)
	{
		time++;
		if(n==0&&k==0)
			break;
		for(i=0;i<n;i++)             //初始化
		{
			f[i]=i;
			p[i]=i;
		}
		sum=n;
		for(i=0;i<k;i++)
		{
			scanf("%s",s);
			if(s[0]=='M')
			{
				scanf("%d%d",&a,&b);
				fr=find(f[a]);
				ed=find(f[b]);
				if(fr!=ed)
					p[fr]=ed;
			}
			else
			{
				scanf("%d",&a);
				f[a]=sum;           //將f[a]的值賦爲sum，則原來的f[a]在也無法訪問，相當於刪除點，而邊並未剪斷
				p[sum]=sum;          //給新的節點父節點
				sum++;
			}
		}
		sum=0;
		memset(flag,0,sizeof(flag));
		for(j=0;j<n;j++)
		{
			fr=find(f[j]);
			if(flag[fr]==0)
			{
				flag[fr]=1;
				sum++;
			}
		}
		printf("Case #%d: %d\n",time,sum);
	}
	return 0;
}