信息奥赛一本通 并查集 团伙(group)

团伙(group)

时间限制: 1000 ms         内存限制: 65536 KB
提交数: 596     通过数: 247 

【题目描述】

在某城市里住着n个人，任何两个认识的人不是朋友就是敌人，而且满足：

1、我朋友的朋友是我的朋友；

2、我敌人的敌人是我的朋友；

所有是朋友的人组成一个团伙。告诉你关于这n个人的m条信息，即某两个人是朋友，或者某两个人是敌人，请你编写一个程序，计算出这个城市最多可能有多少个团伙？

【输入】

第1行为n和m，1<n<1000,1≤m≤100 000；

以下m行，每行为p x y，p的值为0或1，p为0时，表示x和y是朋友，p为1时，表示x和y是敌人。

【输出】

一个整数，表示这n个人最多可能有几个团伙。

【输入样例】

6 4
1 1 4
0 3 5
0 4 6
1 1 2

【输出样例】

3

【算法分析】

我们可以把朋友的节点设为一样，记录在fa[i]中，遇到敌人时，将敌人的敌人放到fa[i]中，所以需要a[i]记录敌人。

扫描二维码关注公众号，回复： 666556 查看本文章

【代码实现】

#include<bits/stdc++.h>
using namespace std;
#define N 10100
int fa[N],a[N];
//fa[i]记录父亲节点：朋友，a[i]记录i的第一个敌人
int find(int x)
{
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=2*n;i++)
        fa[i]=i;
    for(int i=1;i<=m;i++)
    {
        int q,x,y;
        scanf("%d%d%d",&q,&x,&y);
        int r1=find(x),r2=find(y);
         if(q==0)
         {
             fa[r2]=r1;
         }
         else
         {//如果是敌人，那么合并x与y的敌人，y也如此
             if(!a[x]) //如果x无敌人，那么就把y当作他的敌人
                a[x]=y;
                else 
                {
               int r3=find(a[x]);
                fa[r3]=r2;  //敌人的敌人是朋友
                }
                //y同理
            if(!a[y])
                a[y]=x;
            else
                {
                    int r4=find(a[y]);
                    fa[r4]=r1;
                }
         }

    }
    int tot=0;
    for(int i=1;i<=n;i++)
        if(fa[i]==i) tot++;
        cout<<tot<<endl;
    return 0;
}