题目描述
给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
思路
递归即可。
代码
class Solution {
public:
TreeNode* BST(vector<int>& nums, int start, int end){
TreeNode* subtree=nullptr;
if(end-start<2){
subtree=new TreeNode(nums[start]);
if(end!=start)
subtree->right = new TreeNode(nums[end]);
return subtree;
}
int mid=(start+end)/2;
subtree=new TreeNode(nums[mid]);
subtree->left=BST(nums,start,mid-1);
subtree->right=BST(nums,mid+1,end);
return subtree;
}
TreeNode* sortedArrayToBST(vector<int>& nums) {
if(!nums.size())return nullptr;
return BST(nums,0,nums.size()-1);
}
};