洛谷 P1892 [BOI2003]团伙

题目描述

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只有一行，表示最大可能的团伙数。

样例

输入

E
F
F
E  

输出

解析

基础并查集，如果是敌人那每次合并的时候就分别遍历一下对方的敌人后合并，如果是朋友的话就直接合并。

代码

#include<bits/stdc++.h>
using namespace std;

int p[];
int fight[][];
int n,m;

void clean()
{
    for(int i=;i<=n;i++)
    {
        p[i]=i;
    }
}

int getf(int v)
{
    if(p[v]==v) return v;
    else
    {
        p[v]=getf(p[v]);
        return p[v];
    }
}

int mer(int a,int b)
{
    int af=getf(a);
    int bf=getf(b);
    if(af!=bf)
    {
        p[bf]=af;
    }
}

int main()
{
    cin>>n
    >>m;
    clean();
    for(int i=;i<=m;i++)
    {
        char relation;
        int a,b;
        cin>>relation>>a>>b;
        if(relation=='F')
        {
            mer(a,b);
        }
        else
        {
            fight[a][b]=;
            fight[b][a]=;//注意双向操作
            for(int j=;j<=n;j++)
            {
                if(fight[b][j]==)
                {
                    mer(a,j);
                }
                if(fight[a][j]==)
                {
                    mer(b,j);
                }
            }//a,b是敌人的话要分别找a的敌人和b的敌人
        }
    }
    int ans=;
    for(int i=;i<=n;i++)
    {
        if(p[i]==i) ans++;
    }
    cout<<ans<<endl;
    return ;
}