[BOI2003]团伙
题目描述
1920年的芝加哥,出现了一群强盗。如果两个强盗遇上了,那么他们要么是朋友,要么是敌人。而且有一点是肯定的,就是:
我朋友的朋友是我的朋友;
我敌人的敌人也是我的朋友。
两个强盗是同一团伙的条件是当且仅当他们是朋友。现在给你一些关于强盗们的信息,问你最多有多少个强盗团伙。
输入输出格式
输入格式:
输入文件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是敌人。输入数据保证不会产生信息的矛盾。
输出格式:
输出文件gangs.out只有一行,表示最大可能的团伙数。
输入输出样例
输入样例#1:
6
4
E 1 4
F 3 5
F 4 6
E 1 2
输出样例#1:
3
这是一道很简单的并查集题目
首先是常规的输入和初始化
int n,m;
cin>>n>>m;//n group m relationship
for(int i=1;i<=n;i++)
{
fu[i]=i;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
guan[i][j]=inf;
}
两个团伙之间是朋友 则进行合并操作
if(x=='F'){
bing(y,z);
}//合并 y和z 团伙
两个团伙之间是敌人 则分别查找对方的敌人 和自己进行合并
if(x=='E')//y和z时朋友
{
for(int j=1;j<=n;j++)
{
if(guan[y][j]=='E'){
bing(z,j);
guan[z][j]=0;
guan[j][z]=0;
}
if(guan[z][j]=='E'){
bing(y,j);
guan[y][j]=0;
guan[j][y]=0;
}
}
}
}
最后统计一下祖先是本身的团伙个数 那么就是最大团伙数
int sum=0;
for(int i=1;i<=n;i++)
{
if(cha(i)==i){
sum++;
}
}
代码整合一下就是
#include <iostream>
#define inf 0x3f3f3f3f
using namespace std;
int fu[2500];
char guan[2500][2500];
int cha(int x){
if(fu[x]!=x) return cha(fu[x]);
else return fu[x];
}
void bing(int x,int y)
{
int fa=cha(x);
int fb=cha(y);
if(fa!=fb) fu[fa]=fb;
}
int main() {
int n,m;
cin>>n>>m;//n group m relationship
for(int i=1;i<=n;i++)
{
fu[i]=i;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
guan[i][j]=inf;
}
for(int i=1;i<=m;i++)
{
char x;
int y,z;
cin>>x>>y>>z;
guan[y][z]=x;
guan[z][y]=x;
if(x=='F'){
bing(y,z);
}//合并 y和z 团伙
if(x=='E')//y和z时朋友
{
for(int j=1;j<=n;j++)
{
if(guan[y][j]=='E'){
bing(z,j);
guan[z][j]=0;
guan[j][z]=0;
}
if(guan[z][j]=='E'){
bing(y,j);
guan[y][j]=0;
guan[j][y]=0;
}
}
}
}
int sum=0;
for(int i=1;i<=n;i++)
{
if(cha(i)==i){
sum++;
}
}
cout<<sum<<endl;
return 0;
}