TZOJ 5784: 团伙 并查集

描述

 

在某城市里住着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

思路：对于朋友，直接并集，对于敌人，那么需要将所有敌人用二维数组存储起来，a[x][0]的数值代表x的敌人数量，然后让敌人并集

#include<bits/stdc++.h>
using namespace std;
const int N = 1005;
int a[N][N],f[N];
int p,x,y;
int find(int x)
{
    if(f[x]!=x)f[x] = find(f[x]);
    return f[x];
}
void merger(int x,int y)
{
    f[y] = x;
}
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)f[i] = i;
    for(int i=1;i<=m;i++)
    {
        cin>>p>>x>>y;
        if(p==0)
        {
            if(find(x)!=find(y))
                merger(x,y);
        }
        else //a[x][0]代表x有a[x][0]个敌人，遍历敌人数量，然后将敌人和y并集 
        {
            for(int j=1;j<=a[x][0];j++) //y和x的敌人并集 
            {
                int fa = find(y);
                int fb = find(a[x][j]);
                if(fa!=fb)merger(fa,fb);
            }
            for(int j=1;j<=a[y][0];j++) //x和y的敌人并集 
            {
                int fa = find(x);
                int fb = find(a[y][j]);
                if(fa!=fb)merger(fa,fb);
            }
            a[x][++a[x][0]] = y; //将y加入x的敌人 
            a[y][++a[y][0]] = x; //将x加入y的敌人 
            
        }
    }
    int ans = 0;
    for(int i=1;i<=n;i++)
        if(f[i]==i)ans++;
    cout<<ans;
     return 0;
}