二叉树的后序遍历 Binary Tree Postorder Traversal

递归方法:

class Solution {
public:
    vector<int> postorderTraversal(TreeNode *root) {
        vector<int> list;
        if(root == NULL)
            return list;
        postorder(root, list);
        return list;
    }
    
    void postorder(TreeNode *root, vector<int> &list)
    {
        if(root->left != NULL)
            postorder(root->left, list);
        if(root->right != NULL)
            postorder(root->right, list);
        
        list.push_back(root->val);
    }
};

迭代方法:

先让根节点进栈。谁出栈则要遍历谁。那么出栈入栈规则如下:
1、如果当前栈顶有左孩子且刚刚没访问过,则栈顶不出栈,且其左孩子入栈。
2、如果当前栈顶没左孩子或左孩子刚访问过,但有右孩子,则栈顶不出栈,且其右孩子入栈。
3、如果当前栈顶左右孩子都没有,则栈顶出栈。
4、如果当前栈顶有右孩子而且是刚刚访问过的,则栈顶出栈。
5、如果当前栈顶没有右孩子,而左孩子是刚刚访问过的,则栈顶出栈。

代码:
/**
 * Definition for binary tree
 * 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> list;
        if(root == NULL)
            return list;
        
        stack<TreeNode *> s;
        s.push(root);
        TreeNode *pre = NULL;
        while(!s.empty())
        {
            TreeNode *now = s.top();
            if((now->left == NULL && now->right == NULL) || ((pre!=NULL)&&(now->right==pre)) || ((pre!=NULL)&&(now->right==NULL)&&(now->left==pre)))
            {
                list.push_back(now->val);
                pre = now;
                s.pop();
                continue;
            }
            
            if(now->left != NULL && now->left != pre)
            {
                s.push(now->left);
                continue;
            }
            
            if(now->right != NULL)
            {
                s.push(now->right);
                continue;
            }
        }
        return list;
    }
};


你可能感兴趣的:(二叉树的后序遍历 Binary Tree Postorder Traversal)