1. 题目
给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
2. 示例
示例一:
输入: 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]
3. 分析
1.前序和中序可以构造二叉树,中序和后续也可以构造二叉树,但是前序和后序不能构造二叉树。
2.因为有中序就可以确定左子树和右子树区间。而用前序下标可以确定节点在根的左子树还是右子树。所以中序用区间,前序用下标
3.用前序第一个节点做根,再去中序里面找这个根节点,中序里这个节点的左区间就是它的左子树,右区间就是它的右子树,递归左子树与右子树,就可以构建这棵树。递归过程如下:
4. 代码实现
/**
* 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:
TreeNode* _buildTree(vector<int>& preorder,int& pi,vector<int>& inorder,int inBegin,int inEnd)
{
if(inBegin > inEnd)
{
return nullptr;
}
TreeNode* root = new TreeNode(preorder[pi]);
++pi;
int rooti = inBegin;
//在中序数组中找前序第pi个节点
while(rooti <= inEnd)
{
if(root->val == inorder[rooti])
{
break;
}
else
{
++rooti;
}
}
//找到后就递归中序的左子树区间和右子树驱区间
root->left = _buildTree(preorder,pi,inorder,inBegin,rooti-1);
root->right = _buildTree(preorder,pi,inorder,rooti+1,inEnd);
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder)
{
int i = 0;//递归的入参,递归过程中需要修改
return _buildTree(preorder,i,inorder,0,inorder.size()-1);
}
};