1、题目描述
返回与给定先序遍历 preorder 相匹配的二叉搜索树(binary search tree)的根结点。
(回想一下,二叉搜索树是二叉树的一种,其每个节点都满足以下规则,对于 node.left 的任何后代,值总 < node.val,而 node.right 的任何后代,值总 > node.val。此外,先序遍历首先显示节点的值,然后遍历 node.left,接着遍历 node.right。)
输入:[8,5,1,7,10,12]
输出:[8,5,10,1,7,null,12]
2、解题思路
这道题本身也并不难,可以采用遍历先序数组直接插入二叉搜索树的方法,即先初始化二叉树根节点,然后按照二叉搜索树的原则:若节点小于待插入节点则继续查找右子树插入位置,若该节点大于待插入数值,则继续查找左子树插入位置,直到找到一个空节点nullptr。
3、C++代码
class Solution {
public:
TreeNode* bstFromPreorder(vector<int>& preorder) {
TreeNode* result;
int n=preorder.size();
if(n>=1)
result= new TreeNode(preorder[0]);
for(int i=1;i<n;i++)
{
insert_r(result,preorder[i]);
}
return result;
}
void insert_r(TreeNode* p,int n)
{
TreeNode* temp=p;
TreeNode* node =new TreeNode(n);
while(temp->val!=n)
{
if(temp->val>n)
{
if(temp->left==nullptr)
temp->left=node;
temp=temp->left;
}
else
{
if(temp->right==nullptr)
temp->right=node;
temp=temp->right;
}
}
}
};