数据结构源程序 二叉树

0 二叉树的定义

  树是n个元素的有限集合，其相对于线性表不同之处在于每个元素可能有一个至多个后继元素。
树有以下几个特点：
  1.存在一个根节点，没有前驱元素
  2.以根结点外任意结点作为根节点都可以得到一棵新的树
二叉树还多了一个性质：
  3.每个结点至多只存在两个子结点
  因此，这一部分的程序和递归息息相关。

1 相关程序

（1）二叉树的实现

代码如下：

/*二叉树的实现*/
#include<stdio.h>
#include<malloc.h>
#define ELEMTYPE char
#define MAXSIZE 20 
typedef struct nodel
{
	ELEMTYPE data;
	struct nodel *lchild,*rchild;
}BTLINK;

//创建二叉链表存储结构的二叉树
BTLINK *createbt()
{
	BTLINK *q;
	struct nodel *s[MAXSIZE];
	//创建指针数组
	int i,j;
	char x;
	printf("i,x=");
	while(scanf("%d,%c",&i,&x),i!=0&&x!='$')
	{
		if(i > MAXSIZE){
			printf("i must be less than MAXSIZE\n");
			printf("i,x=");
			continue;
		}
		q=(BTLINK*)malloc(sizeof(BTLINK));
		q->data=x;
		q->lchild=NULL;
		q->rchild=NULL;
		//赋值
		 
		s[i]=q;
		if(i!=1)
		{
			j=i/2;
			if(i%2==0)
				s[j]->lchild=q;
			else
				s[j]->rchild=q;
			//s[j]是s[i]的双亲结点 
		}
		printf("i,x=");
	}
	return s[1];
	//返回指向根节点的指针 
}

//先顺序遍历二叉树
void preorder(BTLINK *bt)
{
	if(bt!=NULL)
	{
		printf("%c",bt->data);
		preorder(bt->lchild);
		preorder(bt->rchild);
		//先序 
	}
}
 //统计二叉树结点个数
int countbt(BTLINK *bt)
{
	int lnum,rnum;
	if(bt==NULL)return 0;
	else
	{
		lnum=countbt(bt->lchild);
		rnum=countbt(bt->rchild);
		return lnum+rnum+1;
		//递归，访问到叶子后自下至上依次求结点数 
	}
}

//计算二叉树的深度
int heightbt(BTLINK *bt)
{
	int lh,rh;
	if(bt==NULL)
		return 0;
	else
	{
		lh=heightbt(bt->lchild);
		rh=heightbt(bt->rchild);
		return (lh>rh?lh:rh)+1;
		//递归，访问到叶子后自下至上依次+1。 
	}
}

//统计二叉树单子树结点个数
int onechildnodes(BTLINK *bt)
{
	int ln,rn;
	if(bt==NULL)
		return 0;
	else if(bt->lchild==NULL&&bt->rchild!=NULL)
		return 1 + onechildnodes(bt->rchild);	
		//递归回溯到此处时令值+1，下面同理 
	else if(bt->lchild!=NULL&&bt->rchild==NULL)
		return 1 + onechildnodes(bt->lchild);
	else
	{
		ln=onechildnodes(bt->lchild);
		rn=onechildnodes(bt->rchild);
		//存在左右子树则继续向下
		return ln+rn;
	}
}

//统计二叉树叶子结点个数
int leafnodes(BTLINK *bt)
{
	int ln,rn;
	if(bt==NULL)
		return 0;
	else if(bt->lchild==NULL&&bt->rchild==NULL)
		return 1;
	else
	{
		ln=leafnodes(bt->lchild);
		rn=leafnodes(bt->rchild);
		//存在子树，继续递归 
		return ln+rn;
	}
}

int main()
{
	BTLINK *bt;
	bt=createbt();
	printf("二叉树的先根遍历序列：\n");
	preorder(bt);
	printf("\n二叉树的结点数为：%d\n",countbt(bt));
	printf("二叉树的深度为：%d\n",heightbt(bt));
	printf("二叉树的单子树结点数为：%d\n",onechildnodes(bt)); 
	printf("二叉树的叶子结点数为：%d\n",leafnodes(bt));
	return 0;
}

（2）最优二叉树（赫夫曼树）

  对于n个结点，给予n个权值，将其构造成一棵二叉树。将其中带权路径长度最短的一棵称为最优二叉树，也叫赫夫曼树。
  构造赫夫曼树方法如下：
  （1）已知有n个结点的权值（w₁，w₂，···，w_n）。
  （2）将其中权值最小的两个结点合并为新结点，新结点的权值是这两个结点权值之和。
  （3）若合并后结点个数不为1，回到（2）；否则结束。
程序实现如下：

#include <stdio.h>
#include<conio.h>
#define MAXBIT      100
#define MAXVALUE  10000
#define MAXLEAF     30
#define MAXNODE    MAXLEAF*2 -1

typedef struct 
{
    int bit[MAXBIT];
    int start;
} HCodeType;        /* 编码结构体 */
typedef struct
{
    int weight;
    int parent;
    int lchild;
    int rchild;
} HNodeType;        /* 结点结构体 */

/* 构造一颗哈夫曼树 */
void HuffmanTree (HNodeType HuffNode[MAXNODE],  int n)
{ 
    /* i、j： 循环变量，m1、m2：构造哈夫曼树不同过程中两个最小权值结点的权值，
        x1、x2：构造哈夫曼树不同过程中两个最小权值结点在数组中的序号。*/
    int i, j, m1, m2, x1, x2;
    /* 初始化存放哈夫曼树数组 HuffNode[] 中的结点 */
    for (i=0; i<2*n-1; i++)
    {
        HuffNode[i].weight = 0;
        HuffNode[i].parent =-1;
        HuffNode[i].lchild =-1;
        HuffNode[i].lchild =-1;
    } /* end for */

    /* 输入 n 个叶子结点的权值 */
    for (i=0; i<n; i++)
    {
        printf ("Please input weight of leaf node %d: \n", i);
        scanf ("%d", &HuffNode[i].weight);
    } /* end for */

    /* 循环构造 Huffman 树 */
    for (i=0; i<n-1; i++)
    {
        m1=m2=MAXVALUE;     /* m1、m2中存放两个无父结点且结点权值最小的两个结点 */
        x1=x2=0;
        /* 找出所有结点中权值最小、无父结点的两个结点，并合并之为一颗二叉树 */
        for (j=0; j<n+i; j++)
        {
            if (HuffNode[j].weight < m1 && HuffNode[j].parent==-1)
            {
                m2=m1; 
                x2=x1; 
                m1=HuffNode[j].weight;
                x1=j;
            }
            else if (HuffNode[j].weight < m2 && HuffNode[j].parent==-1)
            {
                m2=HuffNode[j].weight;
                x2=j;
            }
        } /* end for */
            /* 设置找到的两个子结点 x1、x2 的父结点信息 */
        HuffNode[x1].parent  = n+i;
        HuffNode[x2].parent  = n+i;
        HuffNode[n+i].weight = HuffNode[x1].weight + HuffNode[x2].weight;
        HuffNode[n+i].lchild = x1;
        HuffNode[n+i].rchild = x2;

        printf ("x1.weight and x2.weight in round %d: %d, %d\n", i+1, HuffNode[x1].weight, HuffNode[x2].weight);  /* 用于测试 */
        printf ("\n");
    } /* end for */
} /* end HuffmanTree */

int main(void)
{
    HNodeType HuffNode[MAXNODE];            /* 定义一个结点结构体数组 */
    HCodeType HuffCode[MAXLEAF],  cd;       /* 定义一个编码结构体数组， 同时定义一个临时变量来存放求解编码时的信息 */
    int i, j, c, p, n;
    printf ("Please input n:\n");
    scanf ("%d", &n);
    HuffmanTree (HuffNode, n);
    
    for (i=0; i < n; i++)
    {
        cd.start = n-1;
        c = i;
        p = HuffNode[c].parent;
        while (p != -1)   /* 父结点存在 */
        {
            if (HuffNode[p].lchild == c)
                cd.bit[cd.start] = 0;
            else
                cd.bit[cd.start] = 1;
            cd.start--;        /* 求编码的低一位 */
            c=p;                    
            p=HuffNode[c].parent;    /* 设置下一循环条件 */
        } /* end while */
        
        /* 保存求出的每个叶结点的哈夫曼编码和编码的起始位 */
        for (j=cd.start+1; j<n; j++)
        { HuffCode[i].bit[j] = cd.bit[j];}
        HuffCode[i].start = cd.start;
    } /* end for */
    
    /* 输出已保存好的所有存在编码的哈夫曼编码 */
    for (i=0; i<n; i++)
    {
        printf ("%d 's Huffman code is: ", i);
        for (j=HuffCode[i].start+1; j < n; j++)
        {
            printf ("%d", HuffCode[i].bit[j]);
        }
        printf ("\n");
    }
    getch();
    return 0;
}