106. Construct Binary Tree from Inorder and Postorder Traversal

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

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

class Solution {

public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {

        return helper(inorder,0,inorder.size(),postorder,0,postorder.size());
    }
private:
    TreeNode* helper(vector<int>& inorder,int i,int j,vector<int>& postorder,int ii,int jj)
    {
        // 每次取postorder的最后一个值mid,将其作为树的根节点
        // 然后从inroder中找到mid,将其分割成为两部分,左边作为mid的左子树,右边作为mid的右子树
        // tree:     8 4 10 3 6 9 11
        // Inorder   [3 4 6] 8 [9 10 11]
        // postorder [3 6 4]   [9 11 10] 8

        if(i >= j || ii >= jj)
            return NULL;

        int mid = postorder[jj - 1];

        auto f = find(inorder.begin() + i,inorder.begin() + j,mid);

        int dis = f - inorder.begin() - i;

        TreeNode* root = new TreeNode(mid);
        root -> left = helper(inorder,i,i + dis,postorder,ii,ii + dis);
        root -> right = helper(inorder,i + dis + 1,j,postorder,ii + dis,jj - 1);

        return root;

    }
};
posted @ 2016-03-09 16:37  背锅侠  阅读(167)  评论(0编辑  收藏  举报