二叉树的前序、中序、后序遍历 -- 非递归方式实现

目录

前序遍历:根 左 右

实现思想:

需要创建一个栈和一个vector容器

栈用来保存最左路径的节点

vector用来保存遍历的数据

1.首先用一个while循环将二叉树的最左路径节点全部压入栈内,同时由于前序遍历的特性,也将该节点的值同时压入vector容器内

2.循环完成后,栈顶元素为二叉树的最左节点。将该节点获取为当前节点后,对栈顶元素进行移除。

3.访问当前节点的右子树。

4.重复循环,直到当前节点为空,或者栈为空。

实现过程:

cpp 复制代码
class Solution
{
public:
	//前序遍历:根 左 右(非递归)
	vector<int> preorderTraversal(TreeNode* root)
	{
		stack<TreeNode*> st;
		vector<int> vec;
		TreeNode* cur = root;
		while (cur || !st.empty())
		{
			while (cur)
			{
				//将二叉树最左路径压入栈内
				st.push(cur);
				//将最左节点的值压入vector内
				vec.push_back(cur->val);
				cur = cur->left;
			}
			//获取最左节点
			cur = st.top();
			//将最左节点从栈内移除
			st.pop();

			//访问右子树
			cur = cur->right;
		}
		return vec;
	}
};

中序遍历:左 根 右

实现思想:

和前序遍历十分类似,不同之处在于将节点值压入vector的位置不一样。

前序遍历是在将二叉树最左路径节点压入栈的同时,也将该节点的值压入vector。

但是中序遍历是要先从最左节点开始进行,因此将节点值压入vector的操作换到第二步完成。

创建一个栈和一个vector容器

栈用来保存最左路径的节点

vector用来保存遍历的数据

1.首先用一个while循环将二叉树的最左路径节点全部压入栈内。

2.循环完成后,栈顶元素为二叉树的最左节点。将该节点获取为当前节点后,对栈顶元素进行移除,并将该节点的值压入vector容器

3.访问当前节点的右子树。

4.重复循环,直到当前节点为空,或者栈为空。

实现过程:

cpp 复制代码
class Solution
{
public:
	//中序遍历:左 根 右(非递归)
	vector<int> inorderTraversal(TreeNode* root)
	{
		stack<TreeNode*> st;
		vector<int> vec;
		TreeNode* cur = root;
		while (cur || !st.empty())
		{
			while (cur)
			{
				//将二叉树最左路径压入栈内
				st.push(cur);
				cur = cur->left;
			}
			//获取最左节点
			cur = st.top();
			//将最左节点从栈内移除
			st.pop();
			//将最左节点的值压入vector内
			vec.push_back(cur->val);

			//访问右子树
			cur = cur->right;
		}
		return vec;
	}
};

后序遍历:左 右 根

实现思想:

创建一个栈和一个vector容器

栈用来保存最左路径的节点

vector用来保存遍历的数据

创建两个指针

一个指向当前节点、一个指向上一栈顶节点

1.首先用一个while循环将二叉树的最左路径节点全部压入栈内。

2.循环完成后,栈顶元素为二叉树的最左节点。将栈顶元素获取为top节点。

3.判断top节点是否可以压入vector,条件为top节点的右子树为空,或者top节点的右子树已经被压入vector。满足条件后将该top节点压入vector,同时将prev更新。

3.如果不满足条件,则访问top节点的右子树。

4.重复循环,直到当前节点为空,或者栈为空。

实现过程:

cpp 复制代码
class Solution
{
public:
	//后序遍历:左 右 根(非递归)
	vector<int> postorderTraversal(TreeNode* root)
	{
		stack<TreeNode*> st;
		TreeNode* cur = root;
		TreeNode* prev;
		vector<int> vec;
		while (cur || !st.empty())
		{
			while (cur)
			{
				st.push(cur);
				cur = cur->left;
			}
			//获取栈顶元素
			TreeNode* top = st.top();
			//判断当前的节点是否可以遍历
			if (top->right == nullptr || top->right == prev)
			{
				vec.push_back(top->val);
				st.pop();
				prev = top;
			}
			else
				//访问右子树
				cur = top->right;
		}
		return vec;
	}
};
相关推荐
泡沫冰@6 分钟前
数据库(6)
数据结构
枫叶丹417 分钟前
【Qt开发】多元素类控件(三)-> QTreeWidget
开发语言·数据库·c++·qt
晨非辰21 分钟前
【数据结构入坑指南】--《层序分明:堆的实现、排序与TOP-K问题一站式攻克(源码实战)》
c语言·开发语言·数据结构·算法·面试
hansang_IR30 分钟前
【题解】P2217 [HAOI2007] 分割矩阵 [记忆化搜索]
c++·数学·算法·记忆化搜索·深搜
Voyager_42 小时前
算法学习记录03——二叉树学习笔记:从两道题看透后序位置的关键作用
笔记·学习·算法
云泽8083 小时前
函数模板与类模板:C++泛型编程核心解析
java·开发语言·c++
R-G-B7 小时前
【25】MFC入门到精通——MFC静态文本框 中字符串 连续输出 不覆盖先前的文本 换行输出
c++·mfc·mfc静态文本框输出字符串·mfc静态文本框连续输出字符串·mfc静态文本框换行输出字符串
我搞slam7 小时前
快乐数--leetcode
算法·leetcode·哈希算法
WWZZ20258 小时前
快速上手大模型:机器学习3(多元线性回归及梯度、向量化、正规方程)
人工智能·算法·机器学习·机器人·slam·具身感知
东方佑9 小时前
从字符串中提取重复子串的Python算法解析
windows·python·算法