找朋友
用数组来模拟森林
#include<cstdio>
const int n=100;
int father[n];
bool isfather[n];
int findfather(int x)
{
int a=x;
while(x!=father[x])//查找根节点
{
x=father[x];
}
while(a!=father[a])//压缩使指向根结点的结点都直接指向根节点
{
int z=a;
a=father[a];
father[z]=x;
}
return x;
}
void unionx(int a, int b)//如果两个结点根节点不同 则合并
{
int faba=findfather(a);
int fabb=findfather(b);
if(faba!=fabb)
{
father[faba]=fabb;
}
}
void init(int n)
{
for(int i=0;i<n;i++)//初始化,使得初始时所有节点的父节点都是自身
father[i]=i;
isfather[n]=false;
}
int main()
{
int n,m;
int a,b;//暂存好朋友关系
scanf("%d",&n,&m);//输入总人数和好朋友组数
init(n);
for(int i=0;i<m;i++)
{
scanf("%d",&a,&b);
unionx(a,b);
}
for(int i=1;i<=n;i++)
{
isfather[findfather(i)]=true;
}
int ans=0;
for(int i=1;i<n;i++)
{
ans+=isfather[i];
}
printf("%d",ans);
return 0;
}
social clusters
#include <cstdio>
#include<algorithm>
using namespace std;
const int maxn=1000;
int father[maxn];
bool isroot[maxn];
int course[maxn]={0};
int findfather(int x)
{
int a=x;
while(x!=father[x])
{
x=father[x];
}
while(a!=father[a])
{
int z=a;
a=father[a];
father[z]=x;
}
return x;
}
void unionx(int a,int b)
{
int faba=findfather(a);
int fabb=findfather(b);
if(faba!=fabb)
father[fabb]=faba;
}
void init(int n)
{
for(int i=0;i<n;i++)
{
father[i]=i;
}
isroot[n]=false;
}
bool cmp(int a,int b)
{
return a>b;
}
int main ()
{
int n,sum[maxn],temp;
scanf("%d",&n);
init(n);
for(int i=0;i<n;i++)
{
scanf("%d:",&sum[i]);
for(int j=0;j<sum[i];j++)
{
scanf("%d",&temp);
if(course[temp]=0)
{
course[temp]=i;
}
unionx(i,findfather(course[temp]));
}
}
for(int i=1;i<=n;i++)
{
isroot[findfather(i)]++;
}
int ans=0;
for(int i=1;i<=n;i++)
{
if(isroot[i]!=0)
ans++;
}
printf("%d",ans);
sort(isroot+1,isroot+n+1,cmp);
for(int i=1;i<=ans;i++)
{
printf("%d",isroot[i]);
if(i<ans)
printf(" ");
}
}