题目来源:力扣
题目描述:
给你一棵二叉树的根节点 root
,返回其节点值的 后序遍历 。
示例 1:
输入:root = [1,null,2,3] 输出:[3,2,1]
示例 2:
输入:root = [] 输出:[]
示例 3:
输入:root = [1] 输出:[1]
代码实现:
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
TreeNode* prev = NULL;
TreeNode* cur = root;
vector<int> v;
while(cur || !st.empty())
{
while(cur)
{
st.push(cur);
cur=cur->left;
}
TreeNode* top = st.top();
if(top->right == nullptr || top->right == prev)//一个节点右为空,或者上一个访问的节点是右子树的根,说明右子树访问过了,就可以访问当前根节点了
{
prev = top;
v.push_back(top->val);
st.pop();
}
else
{
cur = top->right;
}
}
return v;
}
};
思路:
本题与前序的那道题比较类似
这里大体思路不说了,说一下不一样的地方
后面遍历有一个麻烦的地方
假设我们访问这颗树,我们访问到了3,我们要访问完3的左子树和右子树,然后再访问3,我们的栈里面有8,3,但是我们不知道此时右子树有没有被访问,也就不能访问3,所以我们需要一个prev,如果我们访问的上一个节点是1,也就是3的左子树的根,那么说明3的右子树还没被访问,如果是6,也就是3的右子树的根,说明3的右子树已经被访问了,我们就可以访问3