LeetCode-强盗团伙

1920年的芝加哥，出现了一群强盗。如果两个强盗遇上了，那么他们要么是朋友，要么是敌人。

而且有一点是肯定的，就是：我朋友的朋友是我的朋友；

我敌人的敌人也是我的朋友。两个强盗是同一团伙的条件是当且仅当他们是朋友。

现在给你一些关于强盗们的信息，问你最多有多少个强盗团伙。

输入格式

第一行包含整数 NN，表示强盗的个数（从 11 编号到 NN）。

第二行包含整数 MM，表示关于强盗的信息条数。

接下来 MM 行，每行可能是 F p q 或是 E p q，FF 表示 pp 和 qq 是朋友，EE 表示 pp 和 qq 是敌人。

输入数据保证不会产生信息的矛盾。

输出格式

输出只有一行，表示最大可能的团伙数。

数据范围

2≤N≤10002≤N≤1000,
1≤M≤50001≤M≤5000,
1≤p,q≤N1≤p,q≤N

输入样例

6
4
E 1 4
F 3 5
F 4 6
E 1 2

输出样例

3

 关键思路：利用反集

若1和2是敌人，则1+n和2是朋友，2+n和1是朋友；

若3和1是敌人，则1+n和3是朋友，3+n和1是朋友；

则2和3是朋友。

#include <iostream>
using namespace std;

const int N=2500;
int dp[N];

// int Find(int v){
//     if(dp[v]==v) return v;
//     return dp[v]=Find(dp[v]);
// }

int Find(int v){
    int val=v;
    while(dp[v]!=v) v=dp[v];
    while(val!=v){
        int next=dp[val];
        dp[val]=v;
        val=next;
    }
    return v;
}

void Union(int v1,int v2){
    int m=Find(v1),n=Find(v2);
    if(m!=n) dp[m]=n;
}
int main(int argc,char* argv[]){
    int n,m;
    cin>>n>>m;
    cin.ignore();
    for(int i=1;i<=2*n;++i) dp[i]=i;
    
    char re;
    int k1,k2;
    for(int i=1;i<=m;++i){
        cin>>re>>k1>>k2;//scanf("%c %d %d\n",&re,&k1,&k2);
        if(re=='F') Union(k1,k2);
        else if(re=='E'){
            Union(k1+n,k2);
            Union(k2+n,k1);
        }
    }
    int res=0;
    for(int i=1;i<=n;++i){
        if(dp[i]==i) res++;
    }
    cout<<res<<endl;
    return 0;
}

标签：val,int,团伙,朋友,强盗,LeetCode,dp,Find
来源： https://blog.csdn.net/Jiangtagong/article/details/116097233