洛谷题单指南-集合-P1892 [BOI2003] 团伙
原题链接:https://www.luogu.com.cn/problem/P1892
题意解读:此题与关押罪犯问题非常像,本质上就是要合并所有的朋友。
解题思路:
首先,初始化并查集;
对于每一对人的关系,如果是朋友,直接进行合并;
如果是敌人,先查看双方之前是否有记录其他敌人,如果有,则将一方与另一方的敌人进行合并,如果没有则记录双方互为敌人,便于下一次合并;
最后,统计集合中p[i] == i的数量,即是答案。
100分代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 1005;
int p[N], e[N]; //p:并查集 e:记录每个人的敌人
int n, m;
int find(int x)
{
if(p[x] == x) return x;
return p[x] = find(p[x]);
}
void merge(int x, int y)
{
p[find(x)] = p[find(y)];
}
int main()
{
cin >> n >> m;
char opt;
int x, y;
for(int i = 1; i <= n; i++) p[i] = i; //并查集初始化
while(m--)
{
cin >> opt >> x >> y;
if(opt == 'F') merge(x, y);
else if(opt == 'E')
{
if(e[x]) merge(e[x], y); //如果x有敌人,就把y和x的敌人合并组团
else e[x] = y;
if(e[y]) merge(x, e[y]); //如果y有敌人,就把x和y的敌人合并组团
else e[y] = x;
}
}
int ans = 0;
for(int i = 1; i <= n; i++)
{
if(p[i] == i) ans++;
}
cout << ans;
return 0;
}