问题 F: 团伙(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<cstdio>
#include<cstring>
using namespace std;
const int maxn = 1e3+5;
int ed[maxn][maxn];
int pre[maxn];
int n,m;
int a,b,c;
int findd(int x)//
{
    int r=x;
    while(r!=pre[r])
    r=pre[r];//寻找到根
    int t=x;
    while(t!=pre[t])//路径压缩
    {
        pre[t]=r;
        t=pre[x];
        x=pre[x];
    }
    return r;
   /* return pre[x]==x?x:findd(pre[x]);寻找根*/
 
}
 
void unite(int x, int y)
{
    int fx=findd(x);
    int fy=findd(y);
    if(fy!=fx)
    pre[fy]=fx;
}
int main()
{
    scanf("%d%d",&n,&m);
    memset(ed,0,sizeof(ed));
    while(m--)
    {
        scanf("%d%d%d",&a,&b,&c);
        if(!a)
        ed[b][c]=ed[c][b]=1;
        else
        ed[b][c]=ed[c][b]=2;
    }
    for(int i=1;i<=n;i++)
    pre[i]=i;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(ed[i][j]==0)
            continue;
            else if(ed[i][j]==1)
            {
                unite(i,j);
            }
            else
            {
                for(int k=1;k<=n;k++)
                {
                if(ed[j][k]==2)
                unite(i,k);
                }
            }
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++)
    if(pre[i]==i)ans++;
    printf("%d\n",ans);
    return 0;
}