给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
示例 1:
输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]
示例 2:
输入: preorder = [-1], inorder = [-1]
输出: [-1]
提示:
1 <= preorder.length <= 3000
inorder.length == preorder.length
-3000 <= preorder[i], inorder[i] <= 3000
preorder 和 inorder 均 无重复 元素
inorder 均出现在 preorder
preorder 保证 为二叉树的前序遍历序列
inorder 保证 为二叉树的中序遍历序列来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/construct-binary-tree-from-preorder-and-inorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
前序后后序能够帮我们划分二叉树的根,而我们的中序遍历能够帮我们划分二叉树的左右子树。
思路:前序创建树,中序分隔左右子树
preorder = [3,9,20,15,7]
inorder = [9,3,15,20,7]
从preoder中我们知道3就是跟,然后从inoder中,3左边的都是左子树,3右边的都是右子树。
9的左和9和3之间没有数据了,也就是划分结束了
然后我们看preoder中的9的右边,是20,也就是20是我们3的右子树的根,然后我们从inoder中知道20左边的15是我们20的左子树,20的右边的7是我们20的右子树
所以我们就成功将这棵树构建出来了。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
//这里的prev是通过前序去确定根。
//inbegin和inend是我们的中序的容器的索引范围
TreeNode* _buildTree(vector<int>& preorder, vector<int>& inorder,int& prei,int inbegin,int inend) {
//如果已经无法创建了,就返回空
//子树区间确认是否继续创建子树。
if(inbegin>inend)
{
return nullptr;
}
//这道题还是一个前序遍历
//每创建一个就将prei++
TreeNode* root=new TreeNode(preorder[prei++]);
//分隔中序
int ini=inbegin;
//这里我们给的是闭区间,所以是<=
while(ini<=inend)
{
//我们已经找到了
if(inorder[ini]==root->val)
{
//直接跳出
break;
}
else
{
++ini;
}
}
//这样我们的区间就被划分成了这样三段
//[inbegin,ini-1] ini [ini+1,inend]
//创建左子树
root->left=_buildTree(preorder,inorder,prei,inbegin,ini-1);
//创建右子树
root->right=_buildTree(preorder,inorder,prei,ini+1,inend);
//最后将我们的root,也就是我们新创建的结点返回给上一层递归调用
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int i=0;
//这里我们调用上面的_buildTree函数,最初始的状态,也就是我们的分区是从0到我们的整一个容器的长度-1的范围
return _buildTree(preorder,inorder,i,0,inorder.size()-1);
}
};