2597 团伙（并查集）

题目描述 Description

1920年的芝加哥，出现了一群强盗。如果两个强盗遇上了，那么他们要么是朋友，要么是敌人。而且有一点是肯定的，就是：

我朋友的朋友是我的朋友；

我敌人的敌人也是我的朋友。

两个强盗是同一团伙的条件是当且仅当他们是朋友。现在给你一些关于强盗们的信息，问你最多有多少个强盗团伙。

输入描述 Input Description

输入文件gangs.in的第一行是一个整数N(2<=N<=1000)，表示强盗的个数（从1编号到N）。第二行M(1<=M<=5000)，表示关于强盗的信息条数。以下M行，每行可能是F p q或是E p q（1<=p q<=N），F表示p和q是朋友，E表示p和q是敌人。输入数据保证不会产生信息的矛盾。

输出描述 Output Description

输出文件gangs.out只有一行，表示最大可能的团伙数。

样例输入 Sample Input

6
4
E 1 4
F 3 5
F 4 6
E 1 2

样例输出 Sample Output

数据范围及提示 Data Size & Hint

2<=N<=1000

1<=M<=5000

1<=p q<=N

分类标签 Tags 点此展开

思路：利用e数组记录该强盗的敌人，如果e[x]不为0,则说明x存在敌人为e[x]，此时y就和e[x]为朋友，合并即可。

#include<stdio.h>
#include<iostream>
#include<string>
#include<math.h>
#include<algorithm>
using namespace  std;
int pre[1001],e[1001];
void init(int n)
{
	for(int i=1;i<=n;i++)
	{
		pre[i]=i;
		e[i]=0;
	}
}
int find(int x)
{
	if(pre[x]==x)
	 return x;
	else
	 return pre[x]=find(pre[x]);
}
void join(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	 pre[fx]=fy;
}
int main()
{
	int n;
	cin>>n;
	init(n);
	int t;
	cin>>t;
	while(t--)
	{
		char c;
		int a,b;
		cin>>c>>a>>b;
		if(c=='E')
		{
			if(e[a])
			{
				join(e[a],b);
			}
			if(e[b])
		    {
		    	join(e[b],a);
			}
			e[a]=b;
		}
		else
		{
			join(a,b);
		}
	}
	int ans=0;
	for(int i=1;i<=n;i++)
	{
		if(pre[i]==i)
		 ans++;
	}
	cout<<ans<<endl;
	
	
	
	
	return 0;
}

2597 团伙（并查集）

猜你喜欢