union-find樹結構代碼

union-find樹結構代碼，考慮路徑壓縮和秩啓發式規則。

直接上源代碼：

#include <stdio.h>
#include <malloc.h>
typedef struct Node
{
	int name;//節點名
	//int count;//以此節點爲根的子節點數量
	int father;//父親節點
	int rank;//秩
}TreeNode, *BiTree;
BiTree initializeNode(int num)
{
	BiTree tn=(BiTree)malloc(num*sizeof(TreeNode));
	for (int i=0;i<num;i++)
	{
		tn[i].name=i;
		//tn[i].count=1;
		tn[i].father=i;//父節點爲自己
		tn[i].rank=0;
	}
	return tn;
}

/************************************************************************
找到某個節點的根節點
輸入參數：
	root 根據節點名找到節點編號
	tn：根據節點編號查詢其父節點
	name：節點名
輸出參數：
	index：根結點節點編號
************************************************************************/
int find(int *root,BiTree tn,int name){
	int index=root[name];
	BiTree t=&tn[index];
	if (t->father!=index)
	{
		t->father=find(root,tn,tn[t->father].name);
	}
	return t->father;
}
//兩個根結點合併
void f_union(int *root,BiTree  bt,int name1,int name2,int re_name){
	int index1=find(root,bt,name1);
	int index2=find(root,bt,name2);
	BiTree t1=&bt[index1];
	BiTree t2=&bt[index2];
	if (t1->rank>=t2->rank){
		//如果T1的秩大，即T1的樹高度高
		t2->father=index1;
		if (t1->rank==t2->rank){
			t1->rank++;
		}
		if (re_name>=0)
		{
			t1->name=re_name;
			root[re_name]=index1;
		}
	}
	else{
		t1->father=index2;
		if (re_name>=0)
		{
			t2->name=re_name;
			root[re_name]=index2;
		}
	}
}
void n_union(int *root,BiTree  bt,int name1,int name2){
	f_union(root,bt,name1,name2,-1);
}
void main()
{
	int num=6;
	BiTree tn=initializeNode(num); 
	int root[11]={-1};
	for (int i=0;i<num;i++)
	{
		root[i]=i;
	}
	n_union(root,tn,1,2);
	n_union(root,tn,3,4);
	n_union(root,tn,4,5);
	n_union(root,tn,4,2);
	n_union(root,tn,2,0);


	for (int i=0;i<num;i++)
	{
		printf("index:%d, ",i);
		printf("name:%d, ",tn[i].name);
		printf("rank:%d, ",tn[i].rank);
		printf("father:%d\n",tn[i].father);
	}
	printf("root[name]:index\n");
	for (int i=0;i<12;i++)
	{
		printf("root[%d]:%d\n",i,root[i]);
	}
	printf("\nprint end");
	char ch = getchar();
}