LeetCode 145. 二叉树的后序遍历（Binary Tree Postorder Traversal）

题目描述

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

示例:

输入: [1,null,2,3]
   1
    \
     2
    /
   3 

输出: [3,2,1]

进阶: 递归算法很简单，你可以通过迭代算法完成吗？

解题思路

后序遍历的顺序是左孩子->右孩子->父节点，对于每个遍历到的节点，执行如下操作：

1. 首先对该节点不断沿左孩子方向向下遍历并入栈，直到左孩子为空
2. 取出栈顶节点，此时该节点为树的最左下节点，若其右孩子不为空，则回到步骤1；若为空说明其为叶子节点，将其值输出到结果中，并记录pre为当前节点
3. 在步骤2判断右孩子是否为空时，同时判断当前节点右孩子是否已被输出，如果pre是其右孩子，说明当前节点的所有子节点已访问过，所以不必再向下遍历，直接输出即可

代码

 /**
  * Definition for a binary tree node.
  * struct TreeNode {
  *     int val;
  *     TreeNode *left;
  *     TreeNode *right;
  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
  * };
  */
 class Solution {
 public:
     vector<int> postorderTraversal(TreeNode* root) {
         vector<int> res;
         TreeNode* node = root, *pre = NULL;
         stack<TreeNode*> s;
         while(node || s.size()){
             while(node){
                 s.push(node);
                 node = node->left;
             }
             node = s.top();
             if(node->right && node->right != pre)
                 node = node->right;
             else{
                 res.push_back(node->val);
                 s.pop();
                 pre = node;
                 node = NULL;
             }
         }
         return res;
     }
 };