题目描述:
返回与给定先序遍历 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]
提示:
1 <= preorder.length <= 100
先序 preorder 中的值是不同的。
方法1:
主要思路:
(1)递归建立,利用二叉搜索树前序遍历的特性建立;
/**
* 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* helper(vector<int>&preorder,int left,int right){
if(left>right){//终止条件
return nullptr;
}
TreeNode* root=new TreeNode(preorder[left]);//建立当前结点
int mid=left+1;
while(mid<=right&&preorder[mid]<preorder[left]){//找出分割点的位置
++mid;
}
//对左右部分分别再建立左右子树
root->left=helper(preorder,left+1,mid-1);
root->right=helper(preorder,mid,right);
return root;//返回根节点
}
TreeNode* bstFromPreorder(vector<int>& preorder) {
if(preorder.empty()){//处理特殊的情形
return nullptr;
}
//建立
return helper(preorder,0,preorder.size()-1);
}
};