洛谷OJ - P1892 - 团伙(并查集)

题目描述

1920年的芝加哥,出现了一群强盗。如果两个强盗遇上了,那么他们要么是朋友,要么是敌人。而且有一点是肯定的,就是:我朋友的朋友是我的朋友;我敌人的敌人也是我的朋友。两个强盗是同一团伙的条件是当且仅当他们是朋友。现在给你一些关于强盗们的信息,问你最多有多少个强盗团伙。
输入
输入文件gangs.in的第一行是一个整数N(2<=N<=1000),表示强盗的个数(从1编号到N)。 第二行M(1<=M<=5000),表示关于强盗的信息条数。 以下M行,每行可能是F p q或是E p q(1<=p q<=N),F表示p和q是朋友,E表示p和q是敌人。输入数据保证不会产生信息的矛盾。
输出
输出文件gangs.out只有一行,表示最大可能的团伙数。
样例输入
6
4
E 1 4
F 3 5
F 4 6
E 1 2
样例输出
3
题目思路

我们用[x,n)表示朋友的区间,[n,n+n)表示敌人的区间。
输入x和y,
1:如果这两个是朋友关系,则直接合并他们。
2:如果是敌对关系,那么x要和y的敌人合并,y也要和x的敌人合并。

题目代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm> 
#define LL long long 
using namespace std;

int p[2010];

int find(int x){
    return p[x] == x ? x : p[x] = find(p[x]);
} 

int x, y, n, m, cnt;
char c;
int main(){

    for(int i = 0; i <= 2010; i++) p[i] = i;
    cnt = 0; 
    
    scanf("%d%d",&n,&m);
    
    for(int i = 1; i <= m; i++){
    
        cin>>c>>x>>y;
    
        if(c == 'F'){
            p[find(x)] = find(y); 
        }else{
            p[find(x+n)] = find(y); // x的敌人与y合并 
            p[find(y+n)] = find(x); // y的敌人与x合并 
        }
    }
    for(int i = 1; i <= n; i++){
        if(p[i] == i)
            cnt++;
    }
    printf("%d\n",cnt);
    
    return 0;
} 
    原文作者:犯罪团伙问题
    原文地址: https://blog.csdn.net/qq_34594236/article/details/72587605
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞