LeetCode145. 二叉树的后序遍历

题目

分析

后序遍历的非递归写法基本上就是按照前序遍历的非递归写出来的。前序遍历：根左右，后序遍历：左右根。如果把前序遍历的结果反转：得到根右左。这就启发我们，前序遍历的非递归代码基本不变，只需要将右孩子先入栈改为左孩子先入栈即可。这样得到的遍历顺序是：根右左。最后反转得到左右根

代码（递归）

class Solution {
public:
    void dfs(TreeNode* root,vector<int>&res){
        if(root == NULL) return;

        dfs(root->left,res);
        dfs(root->right,res);
        res.push_back(root->val);//对根的处理
    }
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int>res;
        dfs(root,res);
        return res;
    }
};

 代码（非递归）

class Solution {
public:
   
    vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*>s;
        vector<int>res;
        s.push(root);

        while(!s.empty()){
            auto t = s.top();
            s.pop();
            if(t!=NULL) res.push_back(t->val);
            else continue;
            if(t->left!=NULL) s.push(t->left);
            if(t->right!=NULL) s.push(t->right);
        }
        reverse(res.begin(),res.end());
        return res;
    }
};