145. 二叉树的后序遍历

题目来源:力扣

题目描述:

给你一棵二叉树的根节点 root ,返回其节点值的 后序遍历 

示例 1:

145. 二叉树的后序遍历_第1张图片

输入:root = [1,null,2,3]
输出:[3,2,1]

示例 2:

输入:root = []
输出:[]

示例 3:

输入:root = [1]
输出:[1]

代码实现:

class Solution {
public:
    vector postorderTraversal(TreeNode* root) {
         stack st;
         TreeNode* prev = NULL;
        TreeNode* cur = root;
        vector 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;
    }
};

思路:

本题与前序的那道题比较类似

 144. 二叉树的前序遍历-C++_KLZUQ的博客-CSDN博客

这里大体思路不说了,说一下不一样的地方

后面遍历有一个麻烦的地方 

145. 二叉树的后序遍历_第2张图片

假设我们访问这颗树,我们访问到了3,我们要访问完3的左子树和右子树,然后再访问3,我们的栈里面有8,3,但是我们不知道此时右子树有没有被访问,也就不能访问3,所以我们需要一个prev,如果我们访问的上一个节点是1,也就是3的左子树的根,那么说明3的右子树还没被访问,如果是6,也就是3的右子树的根,说明3的右子树已经被访问了,我们就可以访问3 

你可能感兴趣的:(算法,leetcode,职场和发展,c++,数据结构)