二叉树(Binary Tree)相关算法的实现

写在前面：

二叉树是比较简单的一种数据结构，理解并熟练掌握其相关算法对于复杂数据结构的学习大有裨益

一.二叉树的创建

[不喜欢理论的点我跳过>>]

所谓的创建二叉树，其实就是让计算机去存储这个特殊的数据结构（特殊在哪里？特殊在它是我们自定义的）

首先，计算机内部存储都是线性的，而我们的树形结构是一种层级的，计算机显然无法理解，计算机能够接受的原始数据类型并不能满足我们的需求

所以，只好自定义一种数据结构来表示层级关系

实际上是要定义结构 + 操作，结构是为操作服务的，举个例子，我们要模拟买票的过程，现有的数据结构无法满足我们的需求（不要提数组…），我们需要的操作可能是:

1.获取站在买票队伍最前面的人；2.把买好票的人踢出队伍；3.第一个人买完票后，他后面的所有人都要“自觉”地向前移动

明确了这三个操作，再根据操作来定义结构，最后我们得到了队列（数组/链表 + 对应的函数）

二叉树也是这样，计算机看到的只是结构 + 操作，结构是Node集合（二叉链表），操作是创建、遍历、查找等等函数

结点：

struct bt
{
	char data;
	struct bt *left;
	struct bt *right;
};

结点就是一个桶，两只手（桶里装数据，两只手伸出去抓左右两个孩子）

操作：

//createBT();
//printBT();
//deleteNode(Node node);
//...

-上面是对二叉树的理解，下面是创建二叉树具体实现-

二叉树的创建过程其实就是遍历过程（此处指递归方式），我们知道二叉树的任何一种遍历方式都可以把树形结构线性化（简单的说就是一组遍历结果可以唯一的表示一颗二叉树），因此可以根据遍历结果来还原一颗二叉树

先序遍历递归建树的具体思路：

1.读入当前根结点的数据；2.如果是空格，则将当前根置为空，否则申请一个新结点，存入数据；3.用当前根结点的左指针和右指针进行递归调用，创建左右子树

语言描述可能不太好懂，代码如下：

struct bt
{
	char data;
	struct bt *left;
	struct bt *right;
};

void createBT(struct bt ** root)
{
	char c;
	c=getchar();
	if(c == ' ')*root=NULL;//若为空格则置空
	else
	{
		*root=(struct bt *)malloc(sizeof(struct bt));//申请新结点
		(*root)->data=c;//存入数据
		createBT(&((*root)->left));//建立当前结点的左子树
		createBT(&((*root)->right));//建立当前结点的右子树
	}
}

例如，如果我们要建立一个二叉树a(b, c)，只要输入它的先序遍历结果ab××c××即可（×表示空格），其余两种建树方式于此类似，不再详述，至于非递归的建树方法参见下面的非递归遍历，非常相似

二.遍历

遍历在实现方式上有递归与非递归两种方式，所谓的非递归其实是由递归转化而来的（手动维护一个栈），开销（内存/时间）上可能非递归的更好一些，毕竟操作系统的栈中维护的信息更多，现场的保存与恢复开销都要更大一些

在遍历顺序上有3种方式：1.先序遍历（根-左-右）；2.中序遍历（左-根-右）；3.后序遍历（左-右-根）

举个例子，二叉树a(b, c(d, e))的三种遍历结果分别是：1.abcde；2.badce；3.bdeca

-下面看看后序遍历的递归与非递归实现，其余的与之类似-

后序遍历递归：

void postOrder(struct bt * root)
{
	if(root == NULL)return;
	else
	{
		postOrder(root->left);
		postOrder(root->right);
		putchar(root->data);
	}
}

后序遍历非递归：

void postOrder(struct st* root)
{
	struct st* stack[100];//声明结点栈
	int top=-1;//栈顶索引
	struct bt* p=root;//当前结点（present）
	struct bt* q=NULL;//上一次处理的结点
	while(p!=NULL||top!=-1)
	{
		for(;p!=NULL;p=p->left)stack[++top]=p;//遍历左子树
		if(top!=-1)
		{
			p=stack[top];
			if(p->right==NULL||p->right==q)//无右孩子，或右孩子已经遍历过
			{
				putchar(p->data);//输出根结点
				q=p;
				p=stack[top];
				top--;
				p=NULL;
			}
			else p=p->right;//遍历右子树
		}
	}
}

为了描述地更清晰，上面直接实现了栈的操作，当然，更规范的做法是将栈作为一个独立的数据结构封装起来，在我们的函数中调用栈提供的操作函数来进行相关操作

三.输出叶结点

检索特定结点的一系列操作都是建立在遍历的基础上的，输出叶结点就是一个例子，叶结点满足的条件是左右孩子都为空，我们只要在遍历中添加这样的判断条件就可以了

//此处采用先序遍历
void printLeaves(struct bt* root)
{
	if(root == NULL)return;
	else
	{
		if(root->left == NULL&&root->right == NULL)putchar(root->data);
		else
		{
			printLeaves(root->left);
			printLeaves(root->right);
		}
	}
}

于此类似的操作有，输出二叉树中满足一定条件的结点，删除指定结点，在指定位置添加结点（子树）…都是在遍历的基础上做一些额外的操作

四.计算树的深度

计算树深有多种方式，例如：1.分别计算根下左右子树的高度，二者中的较大的为树深；2.最大递归深度为树深…

我们采用第一种方式，更清晰一些

int btDepth(struct bt* root)
{
	int rd,ld;
	if(root==NULL)return 0;//空树深度为0
	else
	{
		ld=1+btDepth(root->left);//递归进层，深度加1
		rd=1+btDepth(root->right);//递归进层，深度加1
		return ld > rd ? ld : rd;//返回最大值
	}
}

五.树形输出

所谓树形输出，即对自然表示的二叉树逆时针旋转90度，其实仍然是在遍历的过程中记录递归层数，以此确定输出结果

//depth表示递归深度，初始值为0
void btOutput(struct bt* root,int depth)
{
	int k;
	if(root==NULL)return;
	else
	{
		btOutput(root->right,depth+1);//遍历右子树
		for(k=0;k<depth;k++)
			printf(" ");//递归层数为几，就输出几个空格（缩进几位）
		putchar(root->data);printf("\n");
		btOutput(root->left,depth+1);//遍历左子树
	}
}
//“右-中-左”的遍历顺序被称为“逆中序”遍历，采用这种顺序是为了符合输出规则（逆时针90度）

六.按层缩进输出

按层缩进输出就像代码编辑器中的自动缩进，从根结点开始逐层缩进，只需要对上面的代码稍作改动就可以实现

//k仍然表示递归深度，初始值为0
void indOutput(struct bt* root,int k)
{
	int i;
	if(root!=NULL)
	{
		for(i=1;i<=k;i++)
			putchar(' ');
		putchar(root->data);putchar('\n');
		indOutput(root->left,k+1);
		indOutput(root->right,k+1);
	}
	else return;
}
//按层缩进输出与树形输出的唯一区别就是遍历方式不同，前者是先序遍历，后者是逆中序遍历

七.按层顺序输出

按层顺序输出与前面提及的两种输出方式看似相似，实则有着很大不同，至少，我们无法简单地套用任何一种遍历过程来完成这个目标。所以，只能维护一个队列来控制遍历顺序

void layerPrint(struct bt* root)
{
	struct bt* queue[100];//声明结点队列
	struct bt* p;//当前结点
	int amount=0,head,tail,j,k;//队列相关属性（元素总数，对头、队尾索引）
	queue[0]=root;
	head=0;
	tail=1;
	amount++;
	while(1)
	{
		j=0;
		for(k=0;k<amount;k++)
		{
			p=queue[head++];//取对头元素
			if(p->left!=NULL)
			{
				queue[tail++]=p->left;//如果有则记录左孩子
				j++;
			}
			if(p->right!=NULL)
			{
				queue[tail++]=p->right;//如果有则记录右孩子
				j++;
			}
			putchar(p->data);//输出当前结点值
		}
		amount=j;//更新计数器
		if(amount==0)break;
	}
}

八.计算从根到指定结点的路径

要记录路径，当然不宜用递归的方式，这里采用后序遍历的非递归实现。为什么选择后序遍历？

因为在这种遍历方式中，某一时刻栈中现有的结点恰恰就是从根结点到当前结点的路径（从栈底到栈顶）。严格地说，此时应该用队列来保存路径，因为栈不支持从栈底到栈顶的出栈操作（这样的小细节就把它忽略好了…）

//参数c为指定结点值
void printPath(struct bt* root,char c)
{
	struct st* stack[100];//声明结点栈
	int top=-1;//栈顶索引
	int i;
	struct bt* p=root;//当前结点
	struct bt* q=NULL;//上一次处理的结点
	while(p!=NULL||top!=-1)
	{
		for(;p!=NULL;p=p->left)stack[++top]=p;//遍历左子树
		if(top!=-1)
		{
			p=stack[top];//获取栈顶元素
			if(p->right==NULL||p->right==q)//如果当前结点没有右孩子或者右孩子刚被访问过
			{
				if(p->data==c)//如果找到则输出路径
				{
					for(i=0;i<=top;i++)
					{
						p=stack[i];
						putchar(p->data);
					}
					printf("\n");
					//此处不跳出循环，因为可能存在不唯一的结点值，遍历整个树，找出所有路径
				}
				q=p;
				p=stack[top];
				top--;
				p=NULL;
			}
			else p=p->right;//遍历右子树
		}
	}
}

九.完整源码与截图示例

源码：

#include<stdio.h>

struct bt
{
	char data;
	struct bt *left;
	struct bt *right;
};

void createBT(struct bt ** root)
{
	char c;
	c=getchar();
	if(c == ' ')*root=NULL;
	else
	{
		*root=(struct bt *)malloc(sizeof(struct bt));
		(*root)->data=c;
		createBT(&((*root)->left));
		createBT(&((*root)->right));
	}
}

void preOrder(struct bt * root)
{
	if(root == NULL)return;
	else
	{
		putchar(root->data);
		preOrder(root->left);
		preOrder(root->right);
	}
}

void inOrder(struct bt * root)
{
	if(root == NULL)return;
	else
	{
		inOrder(root->left);
		putchar(root->data);
		inOrder(root->right);
	}
}

void printLeaves(struct bt* root)
{
	if(root == NULL)return;
	else
	{
		if(root->left == NULL&&root->right == NULL)putchar(root->data);
		else
		{
			printLeaves(root->left);
			printLeaves(root->right);
		}
	}
}

int btDepth(struct bt* root)
{
	int rd,ld;
	if(root==NULL)return 0;
	else
	{
		ld=1+btDepth(root->left);
		rd=1+btDepth(root->right);
		return ld > rd ? ld : rd;
	}
}

void btOutput(struct bt* root,int depth)
{
	int k;
	if(root==NULL)return;
	else
	{
		btOutput(root->right,depth+1);
		for(k=0;k<depth;k++)
			printf(" ");
		putchar(root->data);printf("\n");
		btOutput(root->left,depth+1);
	}
}

void postOrder(struct st* root)
{
	struct st* stack[100];
	int top=-1;
	struct bt* p=root;
	struct bt* q=NULL;
	while(p!=NULL||top!=-1)
	{
		for(;p!=NULL;p=p->left)stack[++top]=p;
		if(top!=-1)
		{
			p=stack[top];
			if(p->right==NULL||p->right==q)
			{
				putchar(p->data);
				q=p;
				p=stack[top];
				top--;
				p=NULL;
			}
			else p=p->right;
		}
	}
}

void printPath(struct bt* root,char c)
{
	struct st* stack[100];
	int top=-1;
	int i;
	struct bt* p=root;
	struct bt* q=NULL;
	while(p!=NULL||top!=-1)
	{
		for(;p!=NULL;p=p->left)stack[++top]=p;
		if(top!=-1)
		{
			p=stack[top];
			if(p->right==NULL||p->right==q)
			{
				if(p->data==c)
				{
					for(i=0;i<=top;i++)
					{
						p=stack[i];
						putchar(p->data);
					}
					printf("\n");
				}
				q=p;
				p=stack[top];
				top--;
				p=NULL;
			}
			else p=p->right;
		}
	}
}

void layerPrint(struct bt* root)
{
	struct bt* queue[100];
	struct bt* p;
	int amount=0,head,tail,j,k;
	queue[0]=root;
	head=0;
	tail=1;
	amount++;
	while(1)
	{
		j=0;
		for(k=0;k<amount;k++)
		{
			p=queue[head++];
			if(p->left!=NULL)
			{
				queue[tail++]=p->left;
				j++;
			}
			if(p->right!=NULL)
			{
				queue[tail++]=p->right;
				j++;
			}
			putchar(p->data);
		}
		amount=j;
		if(amount==0)break;
	}
}

void indOutput(struct bt* root,int k)
{
	int i;
	if(root!=NULL)
	{
		for(i=1;i<=k;i++)
			putchar(' ');
		putchar(root->data);putchar('\n');
		indOutput(root->left,k+1);
		indOutput(root->right,k+1);
	}
	else return;
}

void main()
{
	char c;
	struct bt * root;
	printf("请输入先序遍历结果: ");
	createBT(&root);
	printf("先序遍历(preOrder)[递归]: \n");
	preOrder(root);
	printf("\n中序遍历(inOrder)[递归]: \n");
	inOrder(root);
	printf("\n后序遍历(postOrder)[非递归]: \n");
	postOrder(root);
	printf("\n叶结点(leaves): \n");
	printLeaves(root);
	printf("\n深度(depth): \n");
	printf("%d\n",btDepth(root));
	printf("树形输出(tree output): \n");
	btOutput(root,0);
	printf("缩进输出(indentation output): \n");
	indOutput(root,0);
	printf("请输入目标结点(target node): ");
	getchar();
	c=getchar();
	printf("路径(path): \n");
	printPath(root,c);
	printf("按层输出(layerPrint): \n");
	layerPrint(root);
	printf("\n");
}

截图示例：

一颗较为复杂的二叉树：

其先序遍历结果为：ABD××EG×××C×FH××I××（×表示空，输入的时候要把×换成空格）

把D改成H再试一次（存在重复元素了，因该有两条到H的路径）

发表评论

电子邮件地址不会被公开。 必填项已用*标注

*

code