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

3

数据范围及提示 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;
}

 

    原文作者:犯罪团伙问题
    原文地址: https://blog.csdn.net/weixin_42173193/article/details/82050406
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞