递归 和 非递归 遍历二叉树

1 二叉树结点

2 先序遍历二叉树

3 中序遍历二叉树

4 后序遍历二叉树

5 测试样例

1 二叉树结点

struct BinaryTreeNode

{

int m_nValue;

BinaryTreeNode *m_pLeft;

BinaryTreeNode *m_pRight;

};

2 先序遍历二叉树 

先序遍历二叉树递归算法定义为:

若二叉树为空,则空操作;否则(1)访问根节点;(2)先序遍历左子树;(3)后序遍历右子树;

代码如下:

/* 递归先序遍历 */
void PreOrderTraverseTree(BinaryTreeNode *pRoot)
{
	if (pRoot != NULL)
	{
		cout << pRoot->m_nValue << " ";
		PreOrderTraverseTree(pRoot->m_pLeft);
		PreOrderTraverseTree(pRoot->m_pRight);
	}
}

先序遍历二叉树的非递归算法定义为:

我们首先创建一个堆栈stack<BinaryTreeNode *>nodes,首先让根节点入栈。

然后进入一个while循环,即 while(!nodes.empty()),每次弹出一个节点。遍历该节点的值,然后将其节点的右孩子入栈,再次将其节点的左孩子入栈。

直至while循环结束。

代码如下:

/* 非递归先序遍历 */
void PreOrderTraverseTreeNoRecursive(BinaryTreeNode *pRoot)
{
	if (pRoot == NULL)
	{
		return;
	}

	stack<BinaryTreeNode *> nodes;
	nodes.push(pRoot);
	while (!nodes.empty())
	{
		BinaryTreeNode *pNode = nodes.top();
		nodes.pop();
		cout << pNode->m_nValue << " ";

		if (pNode->m_pRight != NULL)
		{
			nodes.push(pNode->m_pRight);
		}
		if (pNode->m_pLeft != NULL)
		{
			nodes.push(pNode->m_pLeft);
		}
	}
}

3 中序遍历二叉树

中序遍历二叉树的递归算法如下:

若二叉树为空,则空操作;否则(1)中序遍历左子树;(2)访问根节点;(3)中序遍历右子树;

代码如下:

/* 递归中序遍历 */
void InOrderTraverseTree(BinaryTreeNode *pRoot)
{
	if (pRoot != NULL)
	{
		InOrderTraverseTree(pRoot->m_pLeft);
		cout << pRoot->m_nValue << " ";
		InOrderTraverseTree(pRoot->m_pRight);
	}
}

中序遍历二叉树的非递归算法如下:

/* 非递归中序遍历 */
void InOrderTraverseTreeNoRecursive(BinaryTreeNode *pRoot)
{
	if (pRoot == NULL)
	{
		return;
	}

	stack<BinaryTreeNode *> nodes;
	BinaryTreeNode *pCurrent = pRoot;
	while (!nodes.empty() || pCurrent != NULL)
	{
		if (pCurrent != NULL)
		{
			nodes.push(pCurrent);
			pCurrent = pCurrent->m_pLeft;
		}
		else
		{
			pCurrent = nodes.top();
			nodes.pop();
			cout << pCurrent->m_nValue << " ";
			pCurrent = pCurrent->m_pRight;
		}
	}
}

4 后序遍历二叉树

后序遍历二叉树的递归算法如下:

若二叉树为空,则空操作;否则(1)后序遍历左子树;(2)后序遍历右子树;(3)访问根节点;

/* 递归后序遍历 */
void PostOrderTraverseTree(BinaryTreeNode *pRoot)
{
	if (pRoot != NULL)
	{
		PostOrderTraverseTree(pRoot->m_pLeft);
		PostOrderTraverseTree(pRoot->m_pRight);
		cout << pRoot->m_nValue << " ";
	}
}

后序遍历二叉树的非递归算法如下:

/* 非递归后序遍历 */
void PostOrderTraverseTreeNoRecursive(BinaryTreeNode *pRoot)
{
	if (pRoot == NULL)
	{
		return;
	}

	stack<BinaryTreeNode *> traverseNodes;
	stack<BinaryTreeNode *> nodes;
	traverseNodes.push(pRoot);
	while (!traverseNodes.empty())
	{
		BinaryTreeNode *pNode = traverseNodes.top();
		traverseNodes.pop();
		if (pNode->m_pLeft != NULL)
		{
			traverseNodes.push(pNode->m_pLeft);
		}
		if (pNode->m_pRight != NULL)
		{
			traverseNodes.push(pNode->m_pRight);
		}
		nodes.push(pNode);
	}

	while (!nodes.empty())
	{
		BinaryTreeNode *pNode = nodes.top();
		nodes.pop();
		cout << pNode->m_nValue << " ";
	}
}


5 测试样例

文件 BinaryTree.h

struct BinaryTreeNode 
{
	int                    m_nValue; 
	BinaryTreeNode*        m_pLeft;  
	BinaryTreeNode*        m_pRight; 
};

BinaryTreeNode* CreateBinaryTreeNode(int value);
void ConnectTreeNodes(BinaryTreeNode* pParent, BinaryTreeNode* pLeft, BinaryTreeNode* pRight);
void PrintTreeNode(BinaryTreeNode* pNode);
void PrintTree(BinaryTreeNode* pRoot);
void DestroyTree(BinaryTreeNode* pRoot);

文件 BinaryTree.cpp

#include "BinaryTree.h"
#include <cstdio>

BinaryTreeNode* CreateBinaryTreeNode(int value)
{
	BinaryTreeNode* pNode = new BinaryTreeNode();
	pNode->m_nValue = value;
	pNode->m_pLeft = NULL;
	pNode->m_pRight = NULL;

	return pNode;
}

void ConnectTreeNodes(BinaryTreeNode* pParent, BinaryTreeNode* pLeft, BinaryTreeNode* pRight)
{
	if(pParent != NULL)
	{
		pParent->m_pLeft = pLeft;
		pParent->m_pRight = pRight;
	}
}

void PrintTreeNode(BinaryTreeNode* pNode)
{
	if(pNode != NULL)
	{
		printf("value of this node is: %d\n", pNode->m_nValue);

		if(pNode->m_pLeft != NULL)
			printf("value of its left child is: %d.\n", pNode->m_pLeft->m_nValue);
		else
			printf("left child is null.\n");

		if(pNode->m_pRight != NULL)
			printf("value of its right child is: %d.\n", pNode->m_pRight->m_nValue);
		else
			printf("right child is null.\n");
	}
	else
	{
		printf("this node is null.\n");
	}

	printf("\n");
}

void PrintTree(BinaryTreeNode* pRoot)
{
	PrintTreeNode(pRoot);

	if(pRoot != NULL)
	{
		if(pRoot->m_pLeft != NULL)
			PrintTree(pRoot->m_pLeft);

		if(pRoot->m_pRight != NULL)
			PrintTree(pRoot->m_pRight);
	}
}

void DestroyTree(BinaryTreeNode* pRoot)
{
	if(pRoot != NULL)
	{
		BinaryTreeNode* pLeft = pRoot->m_pLeft;
		BinaryTreeNode* pRight = pRoot->m_pRight;

		delete pRoot;
		pRoot = NULL;

		DestroyTree(pLeft);
		DestroyTree(pRight);
	}
}

文件 TraverseTree.cpp

#include "BinaryTree.h"
#include <iostream>
#include <stack>
using namespace std;

/* 递归先序遍历 */
void PreOrderTraverseTree(BinaryTreeNode *pRoot)
{
	if (pRoot != NULL)
	{
		cout << pRoot->m_nValue << " ";
		PreOrderTraverseTree(pRoot->m_pLeft);
		PreOrderTraverseTree(pRoot->m_pRight);
	}
}

/* 非递归先序遍历 */
void PreOrderTraverseTreeNoRecursive(BinaryTreeNode *pRoot)
{
	if (pRoot == NULL)
	{
		return;
	}

	stack<BinaryTreeNode *> nodes;
	nodes.push(pRoot);
	while (!nodes.empty())
	{
		BinaryTreeNode *pNode = nodes.top();
		nodes.pop();
		cout << pNode->m_nValue << " ";

		if (pNode->m_pRight != NULL)
		{
			nodes.push(pNode->m_pRight);
		}
		if (pNode->m_pLeft != NULL)
		{
			nodes.push(pNode->m_pLeft);
		}
	}
}

/* 递归中序遍历 */
void InOrderTraverseTree(BinaryTreeNode *pRoot)
{
	if (pRoot != NULL)
	{
		InOrderTraverseTree(pRoot->m_pLeft);
		cout << pRoot->m_nValue << " ";
		InOrderTraverseTree(pRoot->m_pRight);
	}
}

/* 非递归中序遍历 */
void InOrderTraverseTreeNoRecursive(BinaryTreeNode *pRoot)
{
	if (pRoot == NULL)
	{
		return;
	}

	stack<BinaryTreeNode *> nodes;
	BinaryTreeNode *pCurrent = pRoot;
	while (!nodes.empty() || pCurrent != NULL)
	{
		if (pCurrent != NULL)
		{
			nodes.push(pCurrent);
			pCurrent = pCurrent->m_pLeft;
		}
		else
		{
			pCurrent = nodes.top();
			nodes.pop();
			cout << pCurrent->m_nValue << " ";
			pCurrent = pCurrent->m_pRight;
		}
	}
}

/* 递归后序遍历 */
void PostOrderTraverseTree(BinaryTreeNode *pRoot)
{
	if (pRoot != NULL)
	{
		PostOrderTraverseTree(pRoot->m_pLeft);
		PostOrderTraverseTree(pRoot->m_pRight);
		cout << pRoot->m_nValue << " ";
	}
}

/* 非递归后序遍历 */
void PostOrderTraverseTreeNoRecursive(BinaryTreeNode *pRoot)
{
	if (pRoot == NULL)
	{
		return;
	}

	stack<BinaryTreeNode *> traverseNodes;
	stack<BinaryTreeNode *> nodes;
	traverseNodes.push(pRoot);
	while (!traverseNodes.empty())
	{
		BinaryTreeNode *pNode = traverseNodes.top();
		traverseNodes.pop();
		if (pNode->m_pLeft != NULL)
		{
			traverseNodes.push(pNode->m_pLeft);
		}
		if (pNode->m_pRight != NULL)
		{
			traverseNodes.push(pNode->m_pRight);
		}
		nodes.push(pNode);
	}

	while (!nodes.empty())
	{
		BinaryTreeNode *pNode = nodes.top();
		nodes.pop();
		cout << pNode->m_nValue << " ";
	}
}

void Test(const char *testName, BinaryTreeNode *pRoot)
{
	if (testName != NULL)
	{
		cout << testName << " : " << endl;
	}
	cout << "递归先序遍历:  " ;
	PreOrderTraverseTree(pRoot);
	cout << endl;
	cout << "非递归先序遍历:";
	PreOrderTraverseTreeNoRecursive(pRoot);
	cout << endl;
	cout << "递归中序遍历:  ";
	InOrderTraverseTree(pRoot);
	cout << endl;
	cout << "非递归中序遍历:";
	InOrderTraverseTreeNoRecursive(pRoot);
	cout << endl;
	cout << "递归后序遍历:  ";
	PostOrderTraverseTree(pRoot);
	cout << endl;
	cout << "非递归后序遍历:";
	PostOrderTraverseTreeNoRecursive(pRoot);
	cout << endl;
}
void Test1()
{
	BinaryTreeNode *pNode1 = CreateBinaryTreeNode(1);
	BinaryTreeNode *pNode2 = CreateBinaryTreeNode(2);
	BinaryTreeNode *pNode3 = CreateBinaryTreeNode(3);
	BinaryTreeNode *pNode4 = CreateBinaryTreeNode(4);
	BinaryTreeNode *pNode5 = CreateBinaryTreeNode(5);
	BinaryTreeNode *pNode6 = CreateBinaryTreeNode(6);

	ConnectTreeNodes(pNode1, pNode2, pNode3);
	ConnectTreeNodes(pNode2, pNode4, pNode5);
	ConnectTreeNodes(pNode3, pNode6, NULL);

	Test("Test1", pNode1);

	DestroyTree(pNode1);
}

void Test2()
{
	BinaryTreeNode *pNode1 = CreateBinaryTreeNode(1);
	BinaryTreeNode *pNode2 = CreateBinaryTreeNode(2);
	BinaryTreeNode *pNode3 = CreateBinaryTreeNode(3);
	BinaryTreeNode *pNode4 = CreateBinaryTreeNode(4);
	BinaryTreeNode *pNode5 = CreateBinaryTreeNode(5);
	ConnectTreeNodes(pNode1, pNode2, NULL);
	ConnectTreeNodes(pNode2, pNode3, NULL);
	ConnectTreeNodes(pNode3, pNode4, NULL);
	ConnectTreeNodes(pNode4, pNode5, NULL);

	Test("Test2", pNode1);

	DestroyTree(pNode1);
}

void Test3()
{
	BinaryTreeNode *pNode1 = CreateBinaryTreeNode(1);
	BinaryTreeNode *pNode2 = CreateBinaryTreeNode(2);
	BinaryTreeNode *pNode3 = CreateBinaryTreeNode(3);
	BinaryTreeNode *pNode4 = CreateBinaryTreeNode(4);
	BinaryTreeNode *pNode5 = CreateBinaryTreeNode(5);
	ConnectTreeNodes(pNode1, NULL, pNode2);
	ConnectTreeNodes(pNode2, NULL, pNode3);
	ConnectTreeNodes(pNode3, NULL, pNode4);
	ConnectTreeNodes(pNode4, NULL, pNode5);

	Test("Test3", pNode1);

	DestroyTree(pNode1);
}

void Test4()
{
	BinaryTreeNode *pNode1 = CreateBinaryTreeNode(1);
	Test("Test4", pNode1);
	DestroyTree(pNode1);
}

void Test5()
{
	Test("Test5", NULL);
}

int main()
{
	Test1();
	Test2();
	Test3();
	Test4();
	Test5();
}

测试结果:



遍历二叉树有三种方式,分别是前序遍历、中序遍历和后序遍历。其中,前序遍历指先遍历根节点,再遍历左子树和右子树;中序遍历指先遍历左子树,再遍历根节点和右子树;后序遍历指先遍历左子树和右子树,再遍历根节点。下面是使用递归非递归方式遍历二叉树的示例代码: ```c++ #include <iostream> #include <stack> using namespace std; // 二叉树结点的定义 struct TreeNode { int val; TreeNode* left; TreeNode* right; TreeNode(int x) : val(x), left(NULL), right(NULL) {} }; // 递归前序遍历 void preorderTraversalRecursive(TreeNode* root) { if (root == NULL) return; cout << root->val << " "; preorderTraversalRecursive(root->left); preorderTraversalRecursive(root->right); } // 非递归前序遍历 void preorderTraversalIterative(TreeNode* root) { if (root == NULL) return; stack<TreeNode*> s; s.push(root); while (!s.empty()) { TreeNode* node = s.top(); s.pop(); cout << node->val << " "; if (node->right != NULL) s.push(node->right); if (node->left != NULL) s.push(node->left); } } // 递归中序遍历 void inorderTraversalRecursive(TreeNode* root) { if (root == NULL) return; inorderTraversalRecursive(root->left); cout << root->val << " "; inorderTraversalRecursive(root->right); } // 非递归中序遍历 void inorderTraversalIterative(TreeNode* root) { if (root == NULL) return; stack<TreeNode*> s; TreeNode* node = root; while (node != NULL || !s.empty()) { while (node != NULL) { s.push(node); node = node->left; } node = s.top(); s.pop(); cout << node->val << " "; node = node->right; } } // 递归后序遍历 void postorderTraversalRecursive(TreeNode* root) { if (root == NULL) return; postorderTraversalRecursive(root->left); postorderTraversalRecursive(root->right); cout << root->val << " "; } // 非递归后序遍历 void postorderTraversalIterative(TreeNode* root) { if (root == NULL) return; stack<TreeNode*> s1, s2; s1.push(root); while (!s1.empty()) { TreeNode* node = s1.top(); s1.pop(); s2.push(node); if (node->left != NULL) s1.push(node->left); if (node->right != NULL) s1.push(node->right); } while (!s2.empty()) { cout << s2.top()->val << " "; s2.pop(); } } int main() { // 构造二叉树 TreeNode* root = new TreeNode(1); root->left = new TreeNode(2); root->right = new TreeNode(3); root->left->left = new TreeNode(4); root->left->right = new TreeNode(5); root->right->left = new TreeNode(6); root->right->right = new TreeNode(7); // 递归遍历 cout << "递归前序遍历:"; preorderTraversalRecursive(root); cout << endl; cout << "递归中序遍历:"; inorderTraversalRecursive(root); cout << endl; cout << "递归后序遍历:"; postorderTraversalRecursive(root); cout << endl; // 非递归遍历 cout << "非递归前序遍历:"; preorderTraversalIterative(root); cout << endl; cout << "非递归中序遍历:"; inorderTraversalIterative(root); cout << endl; cout << "非递归后序遍历:"; postorderTraversalIterative(root); cout << endl; return 0; } ``` 注意,非递归方式需要使用栈来存储待遍历的结点,同时也需要注意遍历顺序。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值