树形结构实现并查集

来源:互联网 发布:drgs分组器软件 编辑:程序博客网 时间:2024/05/17 18:13

坑爹货,代码块显示的时候总是出乱码,试了好几次都不行。。

看来今天风水不行,就这样帖上来吧

怒上代码!



#include <iostream>

#define maxsize 20
using namespace std;
int parent[maxsize];

void initial(int parent[])//初始化
{
int i;

for(i=0;i<maxsize;i++)
parent[i]=-1;
}

int find(int x,int parent[])//返回第x个节点所属集合的根节点
{
while(parent[x]>0)
x=parent[x];
return x;
}

void Union(int r1,int r2,int parent[])//将两个集合合并
{
int temp=0;

temp=parent[r1]+parent[r2];
if(parent[r1]>parent[r2])
{//把两个集合中元素个数少的做为个数多的孩子
parent[r1]=r2;
parent[r2]=temp;
}
else
{
parent[r2]=r1;
parent[r1]=temp;
}
}

int main()
{
int i=0,n=0,p=0,k=0;
int r1,r2,x;
int parent[maxsize];

cout<<"请输入初始集合个数:";
cin>>n;

initial(parent);//初始化

cout<<"输入想要合并的次数:";
cin>>k;
for(i=0;i<k;i++)
{
cout<<"输入要合并的两个节点:";
cin>>r1>>r2;
r1=find(r1,parent);
r2=find(r2,parent);
if(r1!=r2)
Union(r1,r2,parent);
}
cout<<"合并之后的集合为:";
for(i=1;i<=n;i++)
cout<<parent[i]<<"  ";

return 0;
}

原创粉丝点击