您的位置:首页 > 其它

LeetCode Construct Binary Tree from Inorder and Postorder Traversal

2014-07-15 18:13 387 查看
class Solution {
public:
TreeNode *buildTree(vector<int> &inorder, vector<int> &postorder) {
int ilen = inorder.size();
int plen = postorder.size();

if (ilen == 0 || plen == 0) return NULL;
return dfs(inorder, postorder, 0, ilen, plen - 1, -1);
}

TreeNode* dfs(vector<int> &inorder, vector<int> &postorder, int is, int ie, int ps, int pe) {

if (is >= ie || ps <= pe) return NULL;
int rval = postorder[ps];
int iroot= is;

while (iroot != ie && inorder[iroot] != rval) iroot++;

int left_len = iroot - is;

TreeNode* nroot = new TreeNode(rval);

nroot->left = dfs(inorder, postorder, is, iroot, pe + left_len, pe);
nroot->right= dfs(inorder, postorder, iroot + 1, ie, ps - 1, pe + left_len);

return nroot;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: