C++实现LeetCode(145.二叉树的后序遍历)
[LeetCode] 145. Binary Tree Postorder Traversal 二叉树的后序遍历
Given a binary tree, return the postorder traversal of its nodes' values.
For example:
Given binary tree {1,#,2,3},
1
\
2
/
3
return [3,2,1].
Note: Recursive solution is trivial, could you do it iteratively?
经典题目，求二叉树的后序遍历的非递归方法，跟前序，中序，层序一样都需要用到栈，后序的顺序是左-右-根，所以当一个结点值被取出来时，它的左右子结点要么不存在，要么已经被访问过了。先将根结点压入栈，然后定义一个辅助结点 head，while 循环的条件是栈不为空，在循环中，首先将栈顶结点t取出来，如果栈顶结点没有左右子结点，或者其左子结点是 head，或者其右子结点是 head 的情况下。将栈顶结点值加入结果 res 中，并将栈顶元素移出栈，然后将 head 指向栈顶元素；否则的话就看如果右子结点不为空，将其加入栈，再看左子结点不为空的话，就加入栈，注意这里先右后左的顺序是因为栈的后入先出的特点，可以使得左子结点先被处理。下面来看为什么是这三个条件呢，首先如果栈顶元素如果没有左右子结点的话，说明其是叶结点，而且入栈顺序保证了左子结点先被处理，所以此时的结点值就可以直接加入结果 res 了，然后移出栈，将 head 指向这个叶结点，这样的话 head 每次就是指向前一个处理过并且加入结果 res 的结点，那么如果栈顶结点的左子结点或者右子结点是 head 的话，说明其子结点已经加入结果 res 了，那么就可以处理当前结点了。
看到这里，大家可能对 head 的作用，以及为何要初始化为 root，还不是很清楚，这里再解释一下。head 是指向上一个被遍历完成的结点，由于后序遍历的顺序是左-右-根，所以一定会一直将结点压入栈，一直到把最左子结点（或是最左子结点的最右子结点）压入栈后，开始进行处理。一旦开始处理了，head 就会被重新赋值。所以 head 初始化值并没有太大的影响，唯一要注意的是不能初始化为空，因为在判断是否打印出当前结点时除了判断是否是叶结点，还要看 head 是否指向其左右子结点，如果 head 指向左子结点，那么右子结点一定为空，因为入栈顺序是根-右-左，不存在右子结点还没处理，就直接去处理根结点了的情况。若 head 指向右子结点，则是正常的左-右-根的处理顺序。那么回过头来在看，若 head 初始化为空，且此时正好左子结点不存在，那么在压入根结点时，head 和左子结点相等就成立了，此时就直接打印根结点了，明显是错的。所以 head 只要不初始化为空，一切都好说，甚至可以新建一个结点也没问题。将 head 初始化为 root，也可以，就算只有一个 root 结点，那么在判定叶结点时就将 root 打印了，然后就跳出 while 循环了，也不会出错。代码如下：
解法一：
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
if (!root) return {};
vector<int> res;
stack<TreeNode*> s{{root}};
TreeNode *head = root;
while (!s.empty()) {
TreeNode *t = s.top();
if ((!t->left && !t->right) || t->left == head || t->right == head) {
res.push_back(t->val);
s.pop();
head = t;
} else {
if (t->right) s.push(t->right);
if (t->left) s.push(t->left);
}
}
return res;
}
};
由于后序遍历的顺序是左-右-根，而先序遍历的顺序是根-左-右，二者其实还是很相近的，可以先在先序遍历的方法上做些小改动，使其遍历顺序变为根-右-左，然后翻转一下，就是左-右-根啦，翻转的方法我们使用反向Q，哦不，是反向加入结果 res，每次都在结果 res 的开头加入结点值，而改变先序遍历的顺序就只要该遍历一下入栈顺序，先左后右，这样出栈处理的时候就是先右后左啦，参见代码如下：
解法二：
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
if (!root) return {};
vector<int> res;
stack<TreeNode*> s{{root}};
while (!s.empty()) {
TreeNode *t = s.top(); s.pop();
res.insert(res.begin(), t->val);
if (t->left) s.push(t->left);
if (t->right) s.push(t->right);
}
return res;
}
};
那么在 Binary Tree Preorder Traversal 中的解法二也可以改动一下变成后序遍历，改动的思路跟上面的解法一样，都是先将先序遍历的根-左-右顺序变为根-右-左，再翻转变为后序遍历的左-右-根，翻转还是改变结果 res 的加入顺序，然后把更新辅助结点p的左右顺序换一下即可，代码如下：
解法三：
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> res;
stack<TreeNode*> s;
TreeNode *p = root;
while (!s.empty() || p) {
if (p) {
s.push(p);
res.insert(res.begin(), p->val);
p = p->right;
} else {
TreeNode *t = s.top(); s.pop();
p = t->left;
}
}
return res;
}
};
论坛上还有一种双栈的解法，其实本质上跟解法二没什么区别，都是利用了改变先序遍历的顺序来实现后序遍历的，参见代码如下：
解法四：
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
if (!root) return {};
vector<int> res;
stack<TreeNode*> s1, s2;
s1.push(root);
while (!s1.empty()) {
TreeNode *t = s1.top(); s1.pop();
s2.push(t);
if (t->left) s1.push(t->left);
if (t->right) s1.push(t->right);
}
while (!s2.empty()) {
res.push_back(s2.top()->val); s2.pop();
}
return res;
}
};
到此这篇关于C++实现LeetCode(145.二叉树的后序遍历)的文章就介绍到这了,更多相关C++实现二叉树的后序遍历内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了C语言之整数划分问题（递归法）实例代码的相关资料,需要的朋友可以参考下2017-02-02
这篇文章主要为大家介绍了C++的友元和内部类，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助2022-01-01
这篇文章主要为大家详细介绍了C语言实现推箱子代码，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-07-07
std::find函数是一个非常实用的通用查找算法,适用于各种场景,本文主要介绍了C++中std::find函数介绍和使用场景,具有一定的参考价值,感兴趣的可以了解一下2024-02-02
这篇文章主要介绍了C++常见异常处理原理及代码示例解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-07-07
C++迭代器介绍（iterator、const_iterator、reverse_interator、const_rev
这篇文章主要介绍了C++迭代器介绍（iterator、const_iterator、reverse_interator、const_reverse_interator）,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-02-02
C++万能头文件它是一个包含了每一个标准库的头文件，接下来通过本文给大家介绍C++中的万能头文件及优缺点,需要的朋友可以参考下2023-02-02
这篇文章主要为大家详细介绍了C++实现简易万年历，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-10-10
这篇文章主要介绍了C语言计算分段函数问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-11-11
在C语言中使用图形库画图需要引入graphics.h库,我们可以通过代码实现画出一个可爱的皮卡丘,将皮卡丘的绘制分为以下部分：耳朵、脸部、眼睛、嘴巴、手臂、腿部、尾巴,下面我们就来学习如何使用C语言编写画皮卡丘的代码2024-01-01
最新评论