二叉树的四种遍历算法

文章目录

    • 前言
    • 递归版本:
    • 非递归版本:

前言

二叉树在作为一种重要的数据结构,它的很多算法的思想在很多地方都用到了,比如说大名鼎鼎的 STL 算法模板,里面的优先队列(priority_queue)、集合(set、map)等等都用到了二叉树里面的思想,如果有兴趣的小伙伴可以去查找一些这些方面的资料。但是我们现在先不讨论那么高深的数据结构,我们先从二叉树的遍历开始:

先来看一下二叉树长什么样子:

二叉树的四种遍历算法_第1张图片

这是百度来的一张二叉树图,我们可以看到, 这棵二叉树一共有 7 个节点,
其中, 0 号节点叫做“根节点”,
下面的 1 号节点和 2 号节点是 0 号节点的子节点,1 号节点为 0 号节点的**“左子节点**, 2 号节点为 0 号节点的 右子节点
同时 1 号节点和 2 号节点又是 3 号节点、 4 号节点和 5 号节点、6号节点的双亲节点
0 号节点有分别以 1 号节点和 2 号节点作为根节点的左右子树
5 号节点和 6 号节点没有子节点(子树),那么它们被称为“叶子节点”。
好了,一些常用的基本概念就到这了,能理解就行,如果还想了解更多专业术语,可以去找一些别的资料。

下面进入正题,二叉树的遍历:

一般来说,二叉树常用的遍历方式有:前序遍历、中序遍历、后序遍历、层序遍历 四种遍历方式,不同的遍历算法,其思想略有不同,我们来看一下这四种遍历方法主要的算法思想:

1、先序遍历二叉树顺序:根节点 --> 左子树 --> 右子树,即先访问根节点,然后是左子树,最后是右子树。
上图中二叉树的前序遍历结果为:0 -> 1 -> 3 -> 4 -> 2 -> 5 -> 6

2、中序遍历二叉树顺序:左子树 --> 根节点 --> 右子树,即先访问左子树,然后是根节点,最后是右子树。
上图中二叉树的中序遍历结果为:3 -> 1 -> 4 -> 0 -> 5 -> 2 -> 6

3、后续遍历二叉树顺序:左子树 --> 右子树 --> 根节点,即先访问左子树,然后是右子树,最后是根节点。
上图中二叉树的后序遍历结果为:3 -> 4 -> 1 -> 5 -> 6 -> 2 -> 0

4、层序遍历二叉树顺序:从最顶层的节点开始,从左往右依次遍历,之后转到第二层,继续从左往右遍历,持续循环,直到所有节点都遍历完成
上图中二叉树的层序遍历结果为:0 -> 1 -> 2 -> 3 -> 4 -> 5 -> 6

下面给出这四种算法思想的伪代码:

递归版本:

前序遍历:

preOrderParse(int n) {
	if(tree[n] == NULL)
		return ; // 如果这个节点不存在,那么结束 
		
	cout << tree[n].w ; // 输出当前节点内容 	
	preOrderParse(tree[n].leftChild); // 递归输出左子树 
	preOrderParse(tree[n].rightChild); // 递归输出右子树 
}

中序遍历:

inOrderParse(int n) {
	if(tree[n] == NULL)
		return ; // 如果这个节点不存在,那么结束 
		
	inOrderParse(tree[n].leftChild); // 递归输出左子树 
	cout << tree[n].w ; // 输出当前节点内容 
	inOrderParse(tree[n].rightChild); // 递归输出右子树 
}

后续遍历:

pastOrderParse(int n) {
	if(tree[n] == NULL)
		return ; // 如果这个节点不存在,那么结束 
		
	pastOrderParse(tree[n].leftChild); // 递归输出左子树 
	pastOrderParse(tree[n].rightChild); // 递归输出右子树 
	cout << tree[n].w ; // 输出当前节点内容 	
}

可以看到前三种遍历都是直接通过递归来完成,用递归遍历二叉树简答方便而且好理解,接下来层序遍历就需要动点脑筋了,我们如何将二叉树一层一层的遍历输出?其实在这里我们要借助一种数据结构来完成:队列。
我们都知道,队列是一种先进先出的数据结构,我们可以先将整颗二叉树的根节点加入队尾,然后循环出队,每次读取对头元素输出并且将队头元素出队,然后将这个输出的元素节点的的左右子树分别依次加入队尾,重复这个循环,知道队列为空的时候结束输出。那么整个二叉树就被我们采用层序遍历的思想输出来了。下面我们看一下上图的二叉树用层序遍历思想的遍历步骤:

二叉树的四种遍历算法_第2张图片

因为笔者不会用 PS,所以用手工代替了,字写的不好,大家多担待,理解过程就行了。好了,对于上图中的步骤,我们用伪代码来实现:

while(!que.empty())  {
	int n = que.front(); // 得到队头元素
	que.pop(); 	// 队头元素出队列 
	// 如果当前节点不为空,那么输出节点的数值,并且在队尾插入左右子节点
	if(tree[n] != NULL) {
		cout << tree[n].w;
		que.push(tree[n].leftChild); 
		que.push(tree[n].rightChild); 
	}
}

Ok,下面来看一下这几个遍历算法的最终代码:

/*
 * 二叉树的四种遍历方式,这里没有采用真实的指针去做,
 * 而是采用数组下标去模拟指针,是一种更加方便快速的方法 
 */
#include 
#include  
using namespace std;
const int N = 10010;
const int INF = -1; // 我们用一个常数来表示当前二叉树节点为空的情况 

struct Node {
	int w; // 当前树节点的值 
	int p; // 当前树节点的双亲所在数组下标 
	int l; // 当前树节点的左子节点所在数组下标 
	int r; // 当前树节点的右子节点所在数组下标 
}; 
Node node[N];

// 按照前序遍历二叉树的顺序输入树节点 
void input(int n) {
	cin >> node[n].w;
	if(node[n].w == INF) { // 输入 -1 代表当前节点所在子二叉树停止输入 
		return ;
	}
	node[n].p = n / 2;
	node[n].l = n * 2;
	node[n].r = n * 2 + 1;
	
	input(n*2);
	input(n*2+1);
}

// 前序遍历二叉树 
void preOrderParse(int n) {
	if(node[n].w == INF) {
		return ;
	}

	cout << node[n].w << " ";
	preOrderParse(node[n].l);
	preOrderParse(node[n].r);
} 

// 中序遍历二叉树 
void inOrderParse(int n) {
	if(node[n].w == INF) {
		return ;
	}
	
	inOrderParse(n*2);
	cout << node[n].w << " ";
	inOrderParse(n*2+1); 
}

// 后续遍历二叉树 
void postOrderParse(int n) {
	if(node[n].w == INF) {
		return ;
	}
	
	postOrderParse(n*2);
	postOrderParse(n*2+1); 
	cout << node[n].w << " ";
} 

/* 
 * 层序遍历二叉树,这里采用的是 c STL 模板的提供的队列(queue),
 * 并没有自己去实现一个队列
 */ 
void sequenceParse() {
	queue<int> que;
	int n = 1;
	que.push(1); // 插入根节点所在数组下标 
	while(!que.empty()) {
		n = que.front();
		que.pop(); // 得到队头元素并且将队头元素出队列 
		// 如果当前节点不为空,那么输出该节点,并且将该节点的左右子节点插入队尾 
		if(node[n].w != INF) { 
			cout << node[n].w << " ";
			que.push(node[n].l);
			que.push(node[n].r);
		}
	}
}

int main() {
	cout << "请以前序遍历的顺序输入二叉树,空节点输入 -1 :" << endl; 
	input(1); // 从下标为 1 开始前序输入二叉树 
	
	cout << "前序遍历:" << endl; 
	preOrderParse(1); 
	cout << endl << "中序遍历:" << endl;
	inOrderParse(1);
	cout << endl << "后序遍历:" << endl;
	postOrderParse(1); 
	cout << endl << "层序遍历:" << endl;
	sequenceParse();
	
	return 0;
} 

结果:

二叉树的四种遍历算法_第3张图片

我们和上面的结果对比一下,完全符合,OK,关于二叉树的四种遍历算法就完成了。上面我们在进行前序、中序和后序遍历的时候采用的是递归的思想进行,而我们知道,在解决相同问题时,递归要比循环的效率低,那么我们就来将上面递归版本的代码改成循环版本吧:

非递归版本:

要将递归改成循环,我们得先知道递归的定义:函数直接 / 间接调用它自己,这两种情况我们分别成为直接递归和间接递归,我们上面看到的就是直接递归。在函数调用过程会有一个函数调用栈,我们可以将每个函数调用过程中的数据看成栈中的一个元素,我们都知道栈是先进后出的一种数据结构,当某个函数被调用的时候就将函数中的相关数据进栈,而当函数结束调用(return )的时候再将相关的数据出栈。我们来看个例子,以上面的前序遍历二叉树的代码为例:

preOrderParse(int n) {
	if(tree[n] == NULL)
		return ; // 如果这个节点不存在,那么结束 
		
	cout << tree[n].w ; // 输出当前节点内容 	
	preOrderParse(tree[n].leftChild); // 递归输出左子树 
	preOrderParse(tree[n].rightChild); // 递归输出右子树 
}

以文章开头给的二叉树的图为输入数据:
二叉树的四种遍历算法_第4张图片
图中绿色数字代表了元素的遍历顺序,其实这种说法并不是特别准确,准确的说是代表了元素进栈和出栈顺序,我们用图来模拟这一过程:
二叉树的四种遍历算法_第5张图片
这个进栈和出栈过程也代表了函数的递归调用的过程,当整个函数调用栈为空的时候结束,即代表遍历完成,那么我们只需要用一个栈来模拟这个函数进栈和出栈的过程就可以模拟整个遍历过程了,伪代码如下:

// 非递归前序遍历二叉树
void preOrder(Node *root) {
	if (root == NULL) {
		return ;
	}
	stack<Node *> stack;
	stack.push(root);
	Node node;
    // 栈非空的情况下
	while (!stack.empty()) {
		// 取得栈顶元素 
		node = stack.top();
		// 栈顶元素出栈
		stack.pop();
		// 遍历得到的栈顶元素
		cout << node->val << " ";
         // 因为栈是先进后出的,所以这里先将右子节点加入栈,再将左子结点加入栈,
		// 达到先遍历左子结点,后遍历右子节点的目的
		if (node->right != NULL) {
			stack.push(node->right);
		}
         // 左子结点加入栈中
		if (node->left != NULL) {
			stack.push(node->left);
		}
	}
}

前序遍历我们已经完成了,那么中序遍历二叉树和后续遍历二叉树的非递归实现就要小伙伴们自己去思考了。

最后给完整的非递归遍历的代码:

/*
 * 二叉树的四种遍历方式,这里没有采用真实的指针去做,
 * 而是采用数组下标去模拟指针,是一种更加方便快速的方法 
 */
#include 
#include 
#include 
using namespace std;
const int N = 10010;
const int INF = -1; // 我们用一个常数来表示当前二叉树节点为空的情况 

struct Node {
	int w; // 当前树节点的值 
	int p; // 当前树节点的双亲所在数组下标 
	int l; // 当前树节点的左子节点所在数组下标 
	int r; // 当前树节点的右子节点所在数组下标 
}; 
Node node[N];

// 按照前序遍历二叉树的顺序输入树节点 
void input(int n) {
	cin >> node[n].w;
	if(node[n].w == INF) { // 输入 -1 代表当前节点所在子二叉树停止输入 
		return ;
	}
	node[n].p = n / 2;
	node[n].l = n * 2;
	node[n].r = n * 2 + 1;
	
	input(n*2);
	input(n*2+1);
}

// 前序遍历二叉树,遍历顺序:根->左->右
void preOrderParse(int root) {
	if (node[root].w == INF) {
		return ;
	}
	
	stack<int> stack;
	stack.push(root);
	Node n;
	while (!stack.empty()) {
		// 取得栈顶元素 
		n = node[stack.top()];
		// 栈顶元素出栈
		stack.pop();
		// 遍历得到的栈顶元素
		cout << n.w << " ";
		// 因为栈是先进后出的,所以这里先将右子节点加入栈,再将左子结点加入栈,
		// 达到先遍历左子结点,后遍历右子节点的目的
		if (node[n.r].w != INF) {
			stack.push(n.r);
		}
		if (node[n.l].w != INF) {
			stack.push(n.l);
		}
	}
}

// 中序遍历二叉树, 遍历顺序:左->根->右
void inOrderParse(int root) {
	if (node[root].w == INF) {
		return ;
	}
	// 标记某个节点是否已经被遍历过
	bool isTravel[N];
	memset(isTravel, 0, N * sizeof(bool));
	stack<int> stack;
	stack.push(root);
	Node n;
	int index;
	while (!stack.empty()) {
		// 取得栈顶元素 
		index = stack.top();
		n = node[index];
		// 如果当前节点的左子结点不为空且没有被遍历过,那么将其加入栈中
		while (node[n.l].w != INF && !isTravel[n.l]) {
			stack.push(n.l);
			index = n.l;
			n = node[index];
		}
		// 栈顶元素出栈
		stack.pop();
		// 遍历得到的节点
		cout << n.w << " ";
		isTravel[index] = true;
		// 如果当前节点的右子结点不为空且没有被遍历过,那么将其加入栈中
		if (node[n.r].w != INF && !isTravel[n.r]) {
			stack.push(n.r);
		}
	}
}

// 后续遍历二叉树, 遍历顺序:左->右->根
void postOrderParse(int root) {
	if (node[root].w == INF) {
		return ;
	}
	bool isTravel[N];
	memset(isTravel, 0, N * sizeof(bool));
	stack<int> stack;
	stack.push(root);
	Node n;
	int index;
	while (!stack.empty()) {
		// 取得栈顶元素 
		index = stack.top();
		n = node[index];
		// 如果当前节点的左子结点不为空且没有被遍历过,那么将其加入栈中
		while (node[n.l].w != INF && !isTravel[n.l]) {
			stack.push(n.l);
			index = n.l;
			n = node[index];
		}
		// 如果当前节点的右子结点不为空且没有被遍历过,那么将其加入栈中,
		// 同时,用 continue 关键字跳过遍历代码,因为右子节点要先于根结点遍历
		if (node[n.r].w != INF && !isTravel[n.r]) {
			stack.push(n.r);
			continue;
		}
		// 栈顶元素出栈
		stack.pop();
		// 遍历得到的节点
		cout << n.w << " ";
		isTravel[index] = true;
		
	}
} 

int main() {
	cout << "请以前序遍历的顺序输入二叉树,空节点输入 -1 :" << endl; 
	input(1); // 从下标为 1 开始前序输入二叉树 
	
	cout << "非递归前序遍历:" << endl; 
	preOrderParse(1); 
	cout << endl << "非递归中序遍历:" << endl;
	inOrderParse(1);
	cout << endl << "非递归后序遍历:" << endl;
	postOrderParse(1); 
	return 0;
} 

好了,我们来看看结果:
二叉树的四种遍历算法_第6张图片
和之前递归遍历的结果一样,好了,二叉树的 4 种遍历方式就介绍到这里了,希望可以帮到你。

如果博客中有什么不正确的地方,还请多多指点,如果觉得我写的不错,请点个赞支持我吧。

谢谢观看。。。

你可能感兴趣的:(树和二叉树)