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

最新推荐文章于 2023-11-03 20:06:50 发布

蓝旭晨枫

最新推荐文章于 2023-11-03 20:06:50 发布

阅读量561

点赞数 1

分类专栏： Algorithm 文章标签：二叉树数据结构算法

本文链接：https://blog.csdn.net/ouyangjinbin/article/details/51079478

版权

Algorithm 专栏收录该内容

46 篇文章 1 订阅

订阅专栏

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;
}