【LeetCode】145. 二叉树的后序遍历-递归和非递归

LeetCode链接:https://leetcode-cn.com/problems/binary-tree-postorder-traversal/submissions/

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

示例:

输入: [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) {}
 * };
 */
class Solution {
public:
    void rec(TreeNode* root,vector<int> &ret){
        if(root){
            rec(root->left,ret);
            rec(root->right,ret);         
            ret.push_back(root->val);
        }
    }
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> ret;
        rec(root,ret);
        return ret;
    }
};

二：非递归

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

思路分析：要想正确的做到后序遍历，要确保根节点在左树和右右树之后push_back()，因此对待任意节点，先push入栈。

若一节点没有左右子树，则直接push_back()该节点。如果有左或右子树，则先遍历子树再push_back()该节点。若不是上

述两种情况，则将该节点的右子树和左子树依次入栈，保证每次取栈顶元素的时候，左孩子在右孩子前面被访问，左孩子

和右孩子都在根结点前面被访问。

/**
 * 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> v;
        if(!root) return v;
        TreeNode* prev = NULL;
        TreeNode* cur = root;
        stack<TreeNode*> s;
        s.push(root);
        
        while(!s.empty())
        {
            cur = s.top();
            if(cur->left == NULL && cur->right == NULL ||
                (prev && (prev == cur->left || prev == cur->right)))
            {
                v.push_back(cur->val);
                s.pop();
                prev = cur;
            }
            else
            {
                if(cur->right)
                    s.push(cur->right);
                if(cur->left)
                    s.push(cur->left);
            }
        }
        return v;
    }
};