CodeVS 2597团伙(并查集)

CodeVS 2597团伙☆☆

题目描述 Description 
1920年的芝加哥,出现了一群强盗。如果两个强盗遇上了,那么他们要么是朋友,要么是敌人。而且有一点是肯定的,就是:我朋友的朋友是我的朋友;我敌人的敌人也是我的朋友。 两个强盗是同一团伙的条件是当且仅当他们是朋友。现在给你一些关于强盗们的信息,问你最多有多少个强盗团伙。

输入描述 Input Description 
输入文件的第一行是一个整数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是敌人。输入数据保证不会产生信息的矛盾。

输出描述 Output Description 
输出文件只有一行,表示最大可能的团伙数。

样例输入 Sample Input 


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

样例输出 Sample Output 
3

数据范围及提示 Data Size & Hint 
2<=N<=1000 
1<=M<=5000 
1<=p,q<=N

分析:通过分析可以发现,一个团伙只能有一个敌人团伙,也就是说团伙是两两为敌的。假如A和B是敌人,那么所有A的其它敌人都是B的朋友,同理B的所有敌人也都是A的朋友,于是就得到上述结论了。基于这样的思想,用f数组代表A所在团伙的代表元,e数组表示A团伙的敌对团伙的代表元,然后进行正常的并集、查询操作就行了。

#include<cstdio>
#define maxn 1010
using namespace std;
int f[maxn],e[maxn],M,N,cnt;
int find(int x)
{
    if(f[x]==x) return x;
    return f[x]=find(f[x]);
}
int main()
{
    int i,p,q;
    char r;
    for(i=1;i<maxn;i++)  f[i]=i;
    scanf("%d%d",&N,&M);
    for(i=1;i<=M;i++)
    {
        scanf("%s%d%d",&r,&p,&q);
        if(r=='F')  f[find(p)]=find(q);
        else
        {
            if(e[p]==0)  e[p]=find(q);
            else
            {
                e[p]=find(e[p]);
                f[e[p]]=find(q);
            }
            if(e[q]==0)  e[q]=find(p);
            else
            {
                e[q]=find(e[q]);
                f[e[q]]=find(p);
            }
        }
    }
    for(i=1;i<=N;i++) cnt+=(f[i]==i);
    printf("%d\n",cnt);
    return 0;
}


你可能感兴趣的:(并查集,codevs,2597团伙)