二叉树遍历(前序、中序、后序、层次)

四种遍历的基本思想:
前序遍历:根节点 --> 左子树 --> 右子树
中序遍历:左子树 --> 根节点 --> 右子树
后序遍历:左子树 --> 右子树 --> 根节点
层次遍历:由上到下,由左到右依次遍历

例子:
二叉树遍历(前序、中序、后序、层次)
前序遍历:1 2 4 5 3 6 7
中序遍历:4 2 5 1 6 3 7
后序遍历:4 5 2 6 7 3 1
层次遍历:1 2 3 4 5 6 7

一、层次遍历
层次遍历需要借助一个队列,代码如下:

//层次遍历
void levelOrder(TreeNode* pRoot) {
	if (!pRoot) return;
	queue<TreeNode*> que;
	que.push(pRoot);
	while (!que.empty()) {
		TreeNode* pNode = NULL;
		pNode = que.front();
		cout << pNode->data << " ";
		que.pop();
		if (pNode->left != NULL)
			que.push(pNode->left);
		if (pNode->right != NULL)
			que.push(pNode->right);
	}
}

二、前序、中序、后序遍历的递归实现

//前序遍历递归实现
void preOrderRecursion(TreeNode* pRoot) {
	if (!pRoot) return;
	cout << pRoot->data << " ";			//根结点
	preOrderRecursion(pRoot->left);		//左子树
	preOrderRecursion(pRoot->right);	//右子树
}
//中序遍历递归实现
void inOrderRecursion(TreeNode* pRoot) {
	if (!pRoot) return;
	inOrderRecursion(pRoot->left);		//左子树
	cout << pRoot->data << " ";			//根结点
	inOrderRecursion(pRoot->right);		//右子树
}
//后序遍历递归实现
void postOrderRecursion(TreeNode* pRoot) {
	if (!pRoot) return;
	postOrderRecursion(pRoot->left);		//左子树
	postOrderRecursion(pRoot->right);		//右子树
	cout << pRoot->data << " ";				//根结点
}

三、前序、中序、后序遍历的非递归实现
能用递归实现的代码,一般都能借助堆栈用非递归实现。

1、前序遍历非递归实现
a、访问根结点它并把入栈,如果根结点左孩子存在,访问左孩子并将其入栈,重复此操作直到当前结点无左孩子;
b、当到达最左端后,栈中结点依次出栈,每出栈一个结点后都判断该结点右孩子是否存在;
c、如果出栈结点的右孩子存在,则以右孩子为根节点重复a操作。

//前序遍历非递归实现
void preOrder(TreeNode* pRoot) {
	if (!pRoot) return;
	stack<TreeNode*> sta;
	TreeNode* p = pRoot;
	while (p != NULL || !sta.empty()) {
		while (p) {
			cout << p->data << " ";
			sta.push(p);
			p = p->left;
		}
		if (!sta.empty()) {
			p = sta.top();
			sta.pop();
			p = p->right;
		}
	}
}

2、中序遍历非递归实现
a、将根结点入栈,如果根结点左孩子存在,将其入栈,重复此操作直到当前结点无左孩子;
b、当到达最左端后,栈中结点依次出栈,每出栈一个结点后都访问它,并判断其右孩子是否存在;
c、如果出栈结点的右孩子存在,则以右孩子为根节点重复a操作。

注:前序遍历和中序遍历非递归实现的思想是一样的,都借助堆栈,唯一区别就是访问结点的时机不同。

//中序遍历非递归实现
void inOrder(TreeNode* pRoot) {
	if (!pRoot) return;
	stack<TreeNode*> sta;
	TreeNode* p = pRoot;
	while (p != NULL || !sta.empty()) {
		while (p) {
			sta.push(p);
			p = p->left;
		}
		if (!sta.empty()) {
			p = sta.top();
			cout << p->data << " ";
			sta.pop();
			p = p->right;
		}
	}
}

3、后序遍历非递归实现
思路:从根结点开始,将所有最左结点入栈,每当一个结点出栈时,都先扫描该节点的右子树,只有当一个结点的左孩子和右孩子均被访问过了,才能访问结点自身。
a、将根结点入栈,如果根结点左孩子存在,将其入栈,重复此操作直到当前结点无左孩子;
b、当到达最左端后,并判断其右孩子是否存在且没有被访问过,若其右孩子存在且为被访问过,重复a操作;
c、若当前结点右孩子不存在或被访问过,则访问当前结点并将其出栈,并将pre指向它(pre标记上一个被访问的结点)。

//后序遍历非递归实现
void postOrder(TreeNode* pRoot) {
	if (!pRoot) return;
	stack<TreeNode*> sta;
	TreeNode* p = pRoot->left, *pre = NULL;
	sta.push(pRoot);
	while (!sta.empty()) {
		while (p) {
			sta.push(p);
			p = p->left;
		}
		TreeNode* pNode = sta.top();
		if (pNode->right && pre != pNode->right) {
			p = pNode->right;
		}
		else {
			cout << pNode->data << " ";
			pre = pNode;
			sta.pop();
		}
	}
}

C++完整代码如下:

#include <iostream>
#include <queue>
#include <stack>
using namespace std;

struct TreeNode {
	int data;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
		data(x), left(NULL), right(NULL) {
	}
};

//层次遍历
void levelOrder(TreeNode* pRoot) {
	if (!pRoot) return;
	queue<TreeNode*> que;
	que.push(pRoot);
	while (!que.empty()) {
		TreeNode* pNode = NULL;
		pNode = que.front();
		cout << pNode->data << " ";
		que.pop();
		if (pNode->left != NULL)
			que.push(pNode->left);
		if (pNode->right != NULL)
			que.push(pNode->right);
	}
}
//前序遍历递归实现
void preOrderRecursion(TreeNode* pRoot) {
	if (!pRoot) return;
	cout << pRoot->data << " ";	//根结点
	preOrderRecursion(pRoot->left);		//左子树
	preOrderRecursion(pRoot->right);		//右子树
}
//中序遍历递归实现
void inOrderRecursion(TreeNode* pRoot) {
	if (!pRoot) return;
	inOrderRecursion(pRoot->left);		//左子树
	cout << pRoot->data << " ";	//根结点
	inOrderRecursion(pRoot->right);		//右子树
}
//后序遍历递归实现
void postOrderRecursion(TreeNode* pRoot) {
	if (!pRoot) return;
	postOrderRecursion(pRoot->left);		//左子树
	postOrderRecursion(pRoot->right);		//右子树
	cout << pRoot->data << " ";				//根结点
}

//前序遍历非递归实现
void preOrder(TreeNode* pRoot) {
	if (!pRoot) return;
	stack<TreeNode*> sta;
	TreeNode* p = pRoot;
	while (p != NULL || !sta.empty()) {
		while (p) {
			cout << p->data << " ";
			sta.push(p);
			p = p->left;
		}
		if (!sta.empty()) {
			p = sta.top();
			sta.pop();
			p = p->right;
		}
	}
}
//中序遍历非递归实现
void inOrder(TreeNode* pRoot) {
	if (!pRoot) return;
	stack<TreeNode*> sta;
	TreeNode* p = pRoot;
	while (p != NULL || !sta.empty()) {
		while (p) {
			sta.push(p);
			p = p->left;
		}
		if (!sta.empty()) {
			p = sta.top();
			cout << p->data << " ";
			sta.pop();
			p = p->right;
		}
	}
}

//后序遍历非递归实现
void postOrder(TreeNode* pRoot) {
	if (!pRoot) return;
	stack<TreeNode*> sta;
	TreeNode* p = pRoot->left, *pre = NULL;
	sta.push(pRoot);
	while (!sta.empty()) {
		while (p) {
			sta.push(p);
			p = p->left;
		}
		TreeNode* pNode = sta.top();
		if (pNode->right && pre != pNode->right) {
			p = pNode->right;
		}
		else {
			cout << pNode->data << " ";
			pre = pNode;
			sta.pop();
		}
	}
}

//二叉树遍历算法
void binaryTreeOrder()
{
	struct TreeNode t1(1), t2(2), t3(3), t4(4), t5(5), t6(6), t7(7);
	t1.left = &t2;
	t1.right = &t3;
	t2.left = &t4;
	t2.right = &t5;
	t3.left = &t6;
	t3.right = &t7;
	printf(" 层次遍历:");
	levelOrder(&t1);
	printf("\n 递归\n");
	printf(" 前序遍历:");
	preOrderRecursion(&t1);
	printf("\n 中序遍历:");
	inOrderRecursion(&t1);
	printf("\n 后序遍历:");
	postOrderRecursion(&t1);
	printf("\n 非递归\n");
	printf(" 前序遍历:");
	preOrder(&t1);
	printf("\n 中序遍历:");
	inOrder(&t1);
	printf("\n 后序遍历:");
	postOrder(&t1);
}

int main()
{
	binaryTreeOrder();
	system("pause");
	return 0;
}

执行结果如下:
二叉树遍历(前序、中序、后序、层次)