【Baltic2003】【BZOJ1370】Gang团伙(并查集,拆点)

problem

  • 给定n个人
  • 朋友的朋友是朋友,敌人的敌人是朋友
  • 朋友之间组成一个团伙,求团伙数

solution

  • 将每个点x拆成两个:x和x+n(分别表示x的朋友和敌人)
  • 如果x和y是朋友,就将x和y合并
  • 如果x和y是敌人,就将x和y+n合并,将y和x+n合并
  • 注意朋友的敌人不一定是敌人,因此如果x和y是朋友,不能将x+n和y+n合并

codes

#include<iostream>
#include<algorithm>
#define maxn 1010
using namespace std;

//UnionFindSet
int fa[maxn<<1];
void init(int n){for(int i = 1; i <= n; i++)fa[i]=i;}
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
void merge(int x, int y){x=find(x);y=find(y);if(x==y)return;fa[x]=y;}

int main(){
    int n, m;
    cin>>n>>m;
    init(maxn<<1);
    for(int i = 1; i <= m; i++){
        char op;  int x, y;
        cin>>op>>x>>y;
        if(op=='F'){
            merge(x,y);
            //merge(x+n,y+n);
        }else{
            merge(x,y+n);
            merge(x+n,y);
        }
    }
    int a[maxn];
    for(int i = 1; i <= n; i++)
        a[i]=find(i);
    sort(a+1,a+n+1);
    int ans = 0;
    for(int i = 1; i <= n; i++)
        if(i==1 || a[i]!=a[i-1])
            ans++;
    cout<<ans<<'\n';
    return 0;
}
    原文作者:犯罪团伙问题
    原文地址: https://blog.csdn.net/qq_33957603/article/details/80525024
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞