【Leetcode】145. Binary Tree Postorder Traversal

题目地址：

https://leetcode.com/problems/binary-tree-postorder-traversal/description/

给定一棵二叉树，求其后序遍历。

法1：DFS。空树则直接返回，否则先递归遍历左子树，接着递归遍历右子树，接着访问树根。代码如下：

/**
 * 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> res;
    dfs(root, res);
    return res;
  }

  void dfs(TreeNode* root, vector<int>& res) {
    if (!root) return;
    dfs(root->left, res);
    dfs(root->right, res);
    res.push_back(root->val);
  }
};

时间复杂度$O(n)$，空间$O(h)$。

法2：栈。从树根开始向下走，能向左走就向左走，走不动了则向右走，直到走到null为止，并把路径节点都入栈。此时栈顶元素左右孩子都是null，从而可以访问，访问完了之后，如果栈空了，那说明当前访问的节点就是树根了，它没有父节点，所以可以退出循环；否则需要看一下它是栈顶的左孩子还是右孩子，如果是左孩子，那就拐到栈顶右孩子那里继续遍历，否则就从栈顶弹出元素继续访问。代码如下：

/**
 * 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> res;
    stack<TreeNode*> stk;
    auto cur = root;
    while (cur || stk.size()) {
      while (cur) {
        stk.push(cur);
        cur = cur->left ? cur->left : cur->right;
      }
      cur = stk.top(); stk.pop();
      res.push_back(cur->val);
      if (stk.size() && stk.top()->left == cur)
        cur = stk.top()->right;
      else cur = nullptr;
    }
    return res;
  }
};

时间复杂度$O(n)$，空间$O(h)$。