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

 

样例

输入

 

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

 

输出

3

解析

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

代码

 

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

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

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