贴一个完整代码:这个程序用于统计n个人,m个两两关系的门派的个数以及每一个门派的人员数量。
代码很简单。也很容易理解
第一种办法及初始化方式:
#include<iostream>
#include<cstring>
using namespace std;
int n, m;//n个人,m个两两团伙,团伙之间存在传递性,即A<->B,B<->C那么A.B.C均是同一门派。
int pre[1000];//用来找掌门用的
int ans;//门派的数量;
void init()
{
memset(pre, -1, sizeof(pre));//初始化为-1;
}
int find(int x)//有路径压缩版
{
if (pre[x]<0)return x;
return find(pre[x]);//找寻根节点
}
void merge(int x, int y)//虚竹和周芷若做朋友
{
int fx = find(x);
int fy = find(y);//一个老大是玄慈,一个是灭绝师太
if (fx != fy)//不是同一个人
{
pre[fx] += pre[fy];//灭绝师太队伍加一
pre[fy] =fx;//方丈很委屈的做了灭绝的小弟 ,靠右原则;//方丈的老大变成灭绝师太
}
}
int main()
{
while (cin >> n >> m)
{
ans = 0;
init();
for (int i = 0; i<m; i++)
{
int x, y;
cin >> x >> y;
merge(x, y);
}
for (int i = 0; i<n; i++)
{
if (pre[i]<0&&pre[i]!=-1)
{
ans++;
//输出每一门派的人员数量
cout << -pre[i] << endl;//第一种办法可以很容易求出各门派人数,而第二种方法很难。
}
}
//输出所有门派的数量;
cout << ans << endl;
}
return 0;
}
第二种办法和初始化方式:
#include<bits/stdc++.h>
using namespace std;
int father[100000];
int find(int x)
{
return x==father[x]?x:find(father[x]);
}
int unions(int a,int b)
{
father[a]=b;
}
int main()
{
int n,m,i;
cin>>n>>m;//n点m边;
for(i=1;i<=n;i++)
father[i]=i;
for(i=1;i<=m;i++)
{
int x,y;
cin>>x>>y;
int r1=find(x);
int r2=find(y);
if(r1!=r2)
{
unions(x,y);
}
}
return 0;
}