给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
示例 1:
输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]
思路:
构造二叉树,第一件事一定是找根节点,然后想办法构造左右子树。
前序遍历序列的第一个值就是根节点,然后在中序遍历序列中寻找这个值,再分左右分别递归构造二叉树。技巧是用四个起点/终点来控制序列的大小。
class Solution {
public:
TreeNode* buildTree(vector<int>& preorder, int pre_start, int pre_end,
vector<int>& inorder, int in_start, int in_end)
{
if (pre_start > pre_end)
{
return nullptr;
}
int val = preorder[pre_start];
TreeNode* root = new TreeNode(val);
int index;
for (index = in_start; index <= in_end; index++)
{
if (inorder[index] == val)
break;
}
int leftsize = index - in_start;
root->left = buildTree(preorder, pre_start + 1, pre_start + leftsize,
inorder, in_start, index - 1);
root->right = buildTree(preorder, pre_start + leftsize + 1, pre_end,
inorder, index + 1, in_end);
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int len = preorder.size();
return buildTree(preorder, 0, len - 1, inorder, 0, len - 1);
}
};