洛谷 P1892 團伙

P1892 團伙

並查集

#include<cstdio>
int fa[2500];//fa[i]表示i的朋友所在集合，fa[i+n]表示i的敵人所在集合
bool boo[2500];
int ans,n,m;
int find(int x)
{
	if(fa[x]!=x)	fa[x]=find(fa[x]);
	return fa[x];
}
void union1(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	fa[fx]=fy;
}//標準的並查集的寫法
int main()
{
	int i,p,q,k;
	char c; 
	scanf("%d%d",&n,&m);
	for(i=1;i<=2*n;i++)
		fa[i]=i;//初始化，最大要初始化的是fa[n+n]即fa[2*n]
	for(i=1;i<=m;i++)
	{
		scanf("\n%c%d%d",&c,&p,&q);
		if(c=='F')
		{
			union1(p,q);
			//union1(p+n,q+n);//別搞錯了，沒有說朋友的敵人一定是敵人，也沒有說自己的敵人與朋友的敵人一定是朋友 
		}
		if(c=='E')
		{
			union1(p,q+n);//合併p的朋友與q的敵人（q的敵人就是p的敵人的敵人） 
			union1(p+n,q);//與上面一行類似
		}
	}
	for(i=1;i<=n;i++)//判斷1-n中有幾個集合，每個集合一個團隊時團隊數量最多（合併集合只可以減少團隊，而要滿足題意最少也要這麼多集合）
	{
		k=find(i);
		if(boo[k]==false)
		{
			boo[k]=true;
			ans++;
		}
	}
	printf("%d",ans);
	return 0;
}