团伙 并查集

问题描述

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

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

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

输入格式

第1行为n和m，N小于1000,M小于5000； 以下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

提示

{1},{2,4,6},{3,5}

限制和约定

时间限制：1s

空间限制：128MB

#include<cstdio>
#include<iostream>
using namespace std;
int n,m,f[10010]={0},zt[10010]={0};
int find(int x)
{
    if(x==f[x]) return x;
    else return f[x]=find(f[x]);
}
void merge(int x,int y)
{
    int rx=find(x);
    int ry=find(y);
    if(rx!=ry) f[rx]=ry;
}
int main()
{
    
    int ch,x,y,ans(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        f[i]=i;
    }
    for(int i=1;i<=m;i++)
    {
        cin>>ch;
        if(ch==1)
        {
            cin>>x>>y;
            if(zt[x]==0) 
            {
            zt[x]=y;
            }
            if(zt[y]==0) 
            {
            zt[y]=x;
            }
            merge(zt[x],y);
            merge(zt[y],x);
        }
        if(ch==0)
        {
            cin>>x>>y;
            merge(x,y);
        }
    }
    for(int i=1;i<=n;i++) if(find(i)==i) ans++;
    cout<<ans;
    return 0;
}