1385：团伙(group)(并查集)

【题目描述】

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

代码：

#include<iostream>
#include<string>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
int n , m;
int f[1005] , book[1005];
void init()
{
    for(int i = 1 ; i <= n ; i++)
        f[i] = i;
}
int getf(int k)
{
    if(f[k] == k) return k;
    else return getf(f[k]);
}
void merge1(int u , int v)
{
    int t1 = getf(u);
    int t2 = getf(v);
    if( t1 != t2)
        f[t2] = t1;
}
int main()
{
    cin>>n>>m;
    init();
    int p , x , y;
    memset(book , 0 , sizeof(book));
    for(int i = 1 ; i <= m ; i++)
    {
        cin>>p>>x>>y;
        if(p == 0)
            merge1(x , y);
        else
        {
            if(book[x] == 0)
                book[x] = y;
            else
                merge1(book[x] , y);
            if(book[y] == 0)
                book[y] = x;
            else
                merge1(book[y] , x);
        }
    }
    int ans = 0;
    for(int i = 1 ; i <= n ; i++)
        if(f[i] == i)
            ans++;
    cout<<ans;
}