LeetCode

Construct Binary Tree from Inorder and Postorder Traversal

2014.1.8 01:16

Given inorder and postorder traversal of a tree, construct the binary tree.

Note:
You may assume that duplicates do not exist in the tree.

Solution:

　　inorder traversal = left, root, right;

　　postorder traversal = left, right, root;

　　Thus the last element in the postorder sequence is the root. Find it in the inorder sequence and you know where the left and right subtrees are. Do this procedure recursively and the job is done.

　　Time and space complexities are both O(n), where n is the number of nodes in the tree. The space complexity comes from the local parameters passed in function calls.

Accepted code:

// 1CE, 1AC, good~
/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode *buildTree(vector<int> &inorder, vector<int> &postorder) {
        // IMPORTANT: Please reset any member data you declared, as
        // the same Solution instance will be reused for each test case.
        return recoverTree(inorder, postorder, 0, inorder.size() - 1, 0, postorder.size() - 1);
    }
private:
    TreeNode *recoverTree(vector<int> &inorder, vector<int> &postorder, int l1, int r1, int l2, int r2) {
        if(l1 > r1){
            return nullptr;
        }
        
        if(l2 > r2){
            return nullptr;
        }
        
        TreeNode *root = new TreeNode(postorder[r2]);
        int i;
        
        for(i = l1; i <= r1; ++i){
            if(inorder[i] == root->val){
                break;
            }
        }
        
        root->left = recoverTree(inorder, postorder, l1, i - 1, l2, l2 + (i - 1 - l1));
        // 1CE here, ， error
        root->right = recoverTree(inorder, postorder, i + 1, r1, r2 - 1 - (r1 - i - 1), r2 - 1);
        
        return root;
    }
};