团伙-并查集

Problem Description:在某城市里住着n个人，任何两个认识的人不是朋友就是敌人，而且满足：
(1)我朋友的朋友是我的朋友；
(2)我敌人的敌人是我的朋友。

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

Input:输入有多组数据，每组数据第1行为n和m，1<n<1000,1<=m<=100000；
以下m行，每行为p、x、y，p的值为0或1，p为0时，表示x和y是朋友，p为1时，表示x和y是敌人。 Output对于每组输入数据，输出一个整数，表示这n个人最多可能有几个团伙。 
Sample Input
6 4
1 1 4
0 3 5
0 4 6
1 1 2

Sample Output

3

Author

HYNU

 

<span style="font-size:14px;">#include<cstdio>
#include<vector>
using namespace std;
#define N 1001
vector<int>num[N];
int f[N],n;

void init()
{
    for(int i=0;i<=n;i++)
        num[i].clear();
    for(int i=1;i<=n;i++)
        f[i]=i;
}

int find(int x)
{
    if(x==f[x]) return x;
    f[x]=find(f[x]);
    return f[x];
}

void Union(int x,int y)
{
    int a=find(x);
    int b=find(y);
    f[b]=a;
}

int main()
{
    int m,p,x,y;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&p,&x,&y);
            if(p==1)
            {
                num[x].push_back(y);
                num[y].push_back(x);
            }
            else
            {
                Union(x,y);
            }
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<num[i].size();j++)
                Union(num[i][0],num[i][j]);
        }
        int ans=0;
        for(int i=1;i<=n;i++)
        {
            if(f[i]==i) ans++;
        }
        printf("%d\n",ans);
    }
    return 0;
}
</span>