C/C++教程

LeetCode-强盗团伙

本文主要是介绍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;
}

 

这篇关于LeetCode-强盗团伙的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!