codevs 2597 团伙 并查集 解题报告

题目描述 Description

1920年的芝加哥，出现了一群强盗。如果两个强盗遇上了，那么他们要么是朋友，要么是敌人。而且有一点是肯定的，就是：

我朋友的朋友是我的朋友；

我敌人的敌人也是我的朋友。

两个强盗是同一团伙的条件是当且仅当他们是朋友。现在给你一些关于强盗们的信息，问你最多有多少个强盗团伙。

输入描述 Input Description

输入文件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是敌人。输入数据保证不会产生信息的矛盾。

输出描述 Output Description

输出文件gangs.out只有一行，表示最大可能的团伙数。

样例输入 Sample Input

6
4
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

思路

拿到题目一看。
就是并查集嘛，没什么大不了的。
但是遇到敌人的关系情况，就记录下来，然后再下一次中更新它，就省去很多判断。
而且也很易懂嘛

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
const int N=10000+5;
int n,m,father[N],ener[N],sum=0;
int getfather(int x)
{
    if (father[x]==x) return x;
    while(father[x]!=x) x=father[x];
    return x;
}
int main()
{
    int i,x,y;
    char ch;
    scanf("%d%d",&n,&m);
    for (i=1;i<=n;i++)
    father[i]=i;
    for (i=1;i<=m;i++)
    {
        cin>>ch>>x>>y;
        if (ch=='F'&&getfather(x)!=getfather(y))
        father[getfather(x)]=getfather(y);
        if (ch=='E')
        {
            if (ener[x]==0) ener[x]=getfather(y);
            else father[getfather(ener[x])]=getfather(y);
            if (ener[y]==0) ener[y]=getfather(x);
            else father[getfather(ener[y])]=getfather(x);
        }
    }
    for (i=1;i<=n;i++)
    if (father[i]==i) sum++;
    printf("%d\n",sum);
    return 0;
}