[BOI 2003]团伙

洛谷 P1892

注意题目里只叙述了两条规则:

• 我的朋友的朋友是我的朋友
• 我的敌人的敌人是我的朋友

而敌人的朋友、朋友的敌人如何并没有交待

---

记录结点 u 的朋友有哪些，敌人有哪些就好了

#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN=1000+1,MAXM=5000+1;
int f[MAXN],u[MAXM],v[MAXM],r[MAXN],h[MAXN];
vector<int>emy[MAXN],fri[MAXN];
inline int find(int x) {
    return f[x]==x?x:find(f[x]);
}
inline void merge(int u,int v) {//启发式合并
    int f1=find(u),f2=find(v);
    if(f1==f2) return;
    if(h[f1]>h[f2]) {
        f[f2]=f1;
    } else if(h[f1]<h[f2]) {
        f[f1]=f2;
    } else {
        f[f1]=f2;
        h[f2]++;
    }
}
inline int read() {
    int c=getchar();
    while(c!='E' && c!='F') c=getchar();
    return c=='E'?0:1;
}
int main() {
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++) f[i]=i;
    for(int i=1;i<=m;i++) {
        //0->enemy,1->firend
        r[i]=read();
        scanf("%d %d",&u[i],&v[i]);
        if(r[i]==1) {
            fri[u[i]].push_back(v[i]);
            fri[v[i]].push_back(u[i]);
        } else {
            emy[u[i]].push_back(v[i]);
            emy[v[i]].push_back(u[i]);
        }
    }
    for(int i=1;i<=m;i++) {
        int a=u[i],b=v[i];
        if(r[i]==1) {
            merge(a,b);//a和b是朋友
            for(int j=0;j<fri[a].size();j++) {//合并b和a的朋友
                merge(b,fri[a][j]);
            }
            for(int j=0;j<fri[b].size();j++) {//合并a和b的朋友
                merge(a,fri[b][j]);
            }
        } else {//a 和 b 是敌人
            for(int j=0;j<emy[b].size();j++) {
                merge(a,emy[b][j]);
            }//合并a和b的敌人
            for(int j=0;j<emy[a].size();j++) {
                merge(b,emy[a][j]);
            }
        }
    }
    int cnt=0;
    for(int i=1;i<=n;i++) {
        if(find(i)==i) cnt++;
    }
    printf("%d",cnt);
    return 0;
}

转载于:https://www.cnblogs.com/Corona09/p/9919821.html