LCR 124.推理二叉树,某二叉树的先序遍历结果记录于整数数组 preorder
,它的中序遍历结果记录于整数数组 inorder
。请根据 preorder
和 inorder
的提示构造出这棵二叉树并返回其根节点。
注意:preorder
和 inorder
中均不含重复数字。
代码如下:
/**
* 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:
//在中序中查找先序的位置,找到返回下标,失败返回-1
int Search(vector<int>& inorder,int startIn,int endIn,int root)//在中序中查找一个结点
{
for(int i=startIn; i<=endIn; i++)
{
if(inorder[i]==root)
return i;
}
return -1;
}
//递归创建二叉树
TreeNode* buildTree(vector<int>& preorder,int startPre,int endPre,vector<int>& inorder,int startIn,int endIn)
{
if(startPre > endPre)//没有数据
return NULL;
TreeNode* p=new TreeNode(preorder[startPre]);//先序的第一个结点是根
int i=Search(inorder,startIn,endIn,preorder[startPre]);//在中序中找根
p->left=buildTree(preorder,startPre+1,startPre+i-startIn,inorder,startIn,i-1);
p->right=buildTree(preorder,startPre+i-startIn+1,endPre,inorder,i+1,endIn);
return p;
}
TreeNode* deduceTree(vector<int>& preorder, vector<int>& inorder) {
return buildTree(preorder,0,preorder.size()-1,inorder,0,inorder.size()-1);
}
};