LeetCode145 二叉树的后序遍历

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

示例:

输入: [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(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
/*
算法思想：
    采用递归的思想，即借助系统栈，效率较低。二叉树的前序遍历规则: 1. 遍历左子树；2. 遍历右子树；3. 访问根结点；
*/
/*
class Solution {
private:
    void rec(TreeNode* root,vector<int> &ret){
        if(root != NULL){            
            rec(root->left,ret);
            rec(root->right,ret);
            ret.push_back(root->val);           
        }
    }
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> ret;
        rec(root,ret);
        return ret;
    }
};
*/
/*
算法思想：
    采用迭代的方法，使用了一个辅助结点p，这种写法其实可以看作是一个模版，对应的还有前序和中序的模版写法，形式很统一，方便于记忆。
    由于后序遍历的顺序是左-右-根，而前序遍历的顺序是根-左-右，二者其实还是很相近的，我们可以先在先序遍历的方法上做些小改动，使其遍历顺序变为根-右-左，然后翻转一下，就是左-右-根了。翻转的方法，是反向加入结果res，每次都在结果res的开头加入结点值，而改变先序遍历的顺序就只要改变一下入栈顺序，先左后右，这样出栈处理的时候就是先右后左了。一定要对比前序遍历记忆！！！
    
拓展：当访问一个结点*p时，栈中结点恰好为*p结点的所有祖先。从栈底到栈底结点再加上*p结点，刚好构成从根节点到*p结点的一条路径。这一特性非常重要，如求根结点到某结点的路径；求两个结点的最近公共祖先；均可用这个思想。
*/
class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> s;
        TreeNode *p = root;
        while (!s.empty() || p) {
            if (p) {
                s.push(p);
                res.insert(res.begin(), p->val);    //反向添加，而前序是正向添加
                p = p->right;
            } else {
                TreeNode *t = s.top(); 
                s.pop();
                p = t->left;
            }
        }
        return res;
    }
};