洛谷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;
}