题目描述
对于一个元素各不相同且按升序排列的有序序列,请编写一个算法,创建一棵高度最小的二叉查找树。
给定一个有序序列int[] vals,请返回创建的二叉查找树的高度。
//当然最简单方法就是返回log2(vals.size())+1
class MinimalBST {
struct TreeNode
{
int val;
TreeNode *left,*right;
TreeNode(int v):val(v),left(nullptr),right(nullptr){}
};
public:
int buildMinimalBST(vector<int> vals) {
if(vals.size()<=0) return 0;
int size=vals.size();
TreeNode *root=CreateBST(vals,0,size-1);
return GetHeight(root);
}
private:
TreeNode* CreateBST(vector<int>&arr,int left,int right)
{
if(left>right)
return nullptr;
int mid=left+(right-left)/2;
TreeNode *root=new TreeNode(arr[mid]);
root->left=CreateBST(arr,left,mid-1);
root->right=CreateBST(arr,mid+1,right);
return root;
}
int GetHeight(TreeNode *root)
{
if(root==nullptr) return 0;
return max(GetHeight(root->left),GetHeight(root->right))+1;
}
};