《剑指offer》
面试题7
i-il+pl是非常重要的一点,只要确定这个右端点其他就好办了。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* dfs(vector<int>& preorder, vector<int>& inorder, int pl, int pr, int il, int ir)
{
if(pl>pr || il>ir)
{
return NULL;
}
TreeNode* root = new TreeNode(preorder[pl]);
if(pl == pr || il == ir)return root;
for(int i=il; i<=ir; i++)
{
if(preorder[pl]==inorder[i])
{
// i-il+pl是非常重要的一点,只要确定这个右端点其他就好办了
root->left = dfs(preorder, inorder, pl+1, i-il+pl, il, i-1);
root->right = dfs(preorder, inorder, i-il+pl+1, pr, i+1, ir );
}
}
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
TreeNode* root ;
root = dfs(preorder, inorder, 0, preorder.size()-1, 0, inorder.size()-1);
return root;
}
};