Given inorder and postorder traversal of a tree, construct the binary tree.
Note:
You may assume that duplicates do not exist in the tree.
/**
* 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) {
// Start typing your C/C++ solution below
// DO NOT write int main() function
return build(inorder,0,inorder.size()-1,postorder,0,postorder.size()-1);
}
TreeNode *build(vector<int> &inorder,int s1,int e1, vector<int> &postorder,int s2,int e2)
{
if(s1 > e1 || s2 > e2) return NULL;
vector<int>::iterator it = find(inorder.begin(),inorder.end(),postorder[e2]);
int index = it - inorder.begin();
TreeNode *root = new TreeNode(postorder[e2]);
TreeNode *left = build(inorder,s1,index-1,postorder,s2,s2+index-s1-1);
TreeNode *right = build(inorder,index+1,e1,postorder,s2+index-s1,e2-1);
root->left = left;
root->right = right;
return root;
}
};
152 milli secs