leetcode 145. 二叉树的后序遍历

leetcode 145. 二叉树的后序遍历

题目详情

题目链接
给定一个二叉树，返回它的 后序 遍历。

• 示例:
  输入: [1,null,2,3]
  1
  \
  2
  /
  3
  输出: [3,2,1]
• 进阶: 递归算法很简单，你可以通过迭代算法完成吗？

我的代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        if (!root) {
            return {};
        }
        vector<int> results;
        stack<TreeNode*> nodes;
        nodes.push(root);
        while (!nodes.empty()) {
           root = nodes.top();
           nodes.pop();
           results.push_back(root->val);
           if (root->left) {
               nodes.push(root->left);
           }
           if (root->right) {
               nodes.push(root->right);
           }
        }
        reverse(results.begin(), results.end());
        return results;
    }
};

我的成绩

执行结果：通过
执行用时：0 ms, 在所有 C++ 提交中击败了100.00%的用户
内存消耗：8.6 MB, 在所有 C++ 提交中击败了9.89%的用户

一些想法

正常的使用迭代来实现后序遍历有些困难，后续遍历的顺序是左->右->中，我们可以反过来遍历中->右->左，也就是变成另一种形式的先序遍历，最后将结果反过来输出。

执行用时为 0 ms 的范例

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> ans;
        if(root==nullptr)return ans;
        stack<TreeNode*> st;
        st.push(root);
        while(!st.empty()){
            TreeNode* node=st.top();
            st.pop();
            ans.push_back(node->val);//访问根
            if(node->left){//后访问左
                st.push(node->left);
            }
            if(node->right){//先访问右
                st.push(node->right);
            }
        }
        reverse(ans.begin(),ans.end());
        return ans;
    }
};

思考

参见官方解答