树的前序遍历非递归、中序遍历非递归、后序遍历非递归

最新推荐文章于 2023-03-15 15:45:48 发布

原创最新推荐文章于 2023-03-15 15:45:48 发布 · 618 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #数据结构 #算法

Algorithm 专栏收录该内容

46 篇文章

订阅专栏

本文详细介绍了二叉树的前序、中序和后序遍历的非递归实现方式，通过使用栈数据结构，实现不依赖递归的遍历过程。

1、二叉树前序遍历非递归

//前序遍历非递归
std::vector<int> PreOrder(TreeNode *root) {
	std::vector<int> ans;
	if (root == NULL)
		return ans;
	std::stack<TreeNode *> stk;
	TreeNode *p = root;
	while (p != NULL || !stk.empty()) {
		if (p != NULL) {
			stk.push(p);
			ans.push_back(p->val);
			p = p->left;
		}
		else {
			p = stk.top();
			stk.pop();
			p = p->right;
		}
	}
	return ans;
}

2、二叉树中序遍历非递归

//中序遍历非递归
std::vector<int> PreOrder(TreeNode *root) {
	std::vector<int> ans;
	if (root == NULL)
		return ans;
	std::stack<TreeNode *> stk;
	TreeNode *p = root;
	while (p != NULL || !stk.empty()) {
		if (p != NULL) {
			stk.push(p);
			p = p->left;
		}
		else {
			p = stk.top();
			stk.pop();
			ans.push_back(p->val);
			p = p->right;
		}
	}
	return ans;
}

3、二叉树后序遍历非递归

//后续遍历非递归
std::vector<int> PreOrder(TreeNode *root) {
	std::vector<int> ans;
	if (root == NULL)
		return ans;
	std::stack<TreeNode *> stk;
	TreeNode *p = root;
	TreeNode *already = NULL; //记录上一次访问过的节点
	while (p != NULL || !stk.empty()) {
		if (p != NULL) {
			stk.push(p);
			p = p->left;
		}
		else {
			p = stk.top();
			//转向右子树
			if (p->right != NULL && p->right != already) {
				p = p->right;
				stk.push(p);
				p = p->left; //再走向左下
			}
			else {
				p = stk.top();
				stk.pop();
				ans.push_back(p->val);
				already = p; //将当前访问过的节点标记为already
				p = NULL; //重要，因为一定是从栈上弹出节点
			}
		}
	}
	return ans;
}