原题链接:[BOI2003]团伙 - 洛谷
解题思路:首先如果为朋友,就直接和普通的并查集操作一样。
如果是敌人且敌人关系不为空,那互相找敌人的敌人所在的集合。因为两个人只是敌对而不与其他人敌对,并不需要并入集合的操作。而如果敌人关系为空,则记录对方为敌人。因为题目只说敌人的敌人就是朋友,数据肯定避免了三个人互相敌对的情况。(何况也没说敌人就不能在一个集合,只要有朋友关系就行嘛XD)
其实是否归属在正确的集合并不影响答案的正确性。因为答案只询问有几个集合,最终只要判断有几个指向自己就可以得到答案了。
AC代码:
#include<bits/stdc++.h>
using namespace std;
int n,m,f[10000],e[10000],ga,gb;//n和m就是题目中原意,f是朋友关系,e是敌人,ga和gb都是在输入关系时缓存用的
char c;//记录输入时的关系
long long ans=0;//记录集合个数,你要是愿意其实unsigned char应该也行
int find_f(int x){//非常传统的并查集
if(f[x]==x) return x;
return f[x]=find_f(f[x]);
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++) f[i]=i;//初始化,每个人都属于自己的集合
for(int i=0;i<m;i++){
cin>>c>>ga>>gb;
if(c=='F'){//如果是朋友就正常的并查集
if(ga>gb){//确保ga比gb小
int buffer;
buffer=ga;
ga=gb;
gb=buffer;
}
f[gb]=find_f(ga);
}
else{//如果是敌人则执行此段代码
if(e[ga]){//如果敌人不为空,说明有加入集合的操作,下面一个if也是
f[gb]=find_f(e[ga]);//找的是敌人的敌人所在的集合
}
if(e[gb]){
f[ga]=find_f(e[gb]);
}
if(!e[gb]) e[gb]=ga;//只有第一次建立敌对才需要记录,后面的敌对直接插入这个敌对所在集合就行了
if(!e[ga]) e[ga]=gb;
}
}
for(int i=1;i<=n;i++) if(f[i]==i) ans++;//找一下f数组中依然指向自己的,就相当于集合的个数
cout<<ans;
return 0;
}