题目描述
给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
链接:https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree
看着这张图才想着取数组中的中间值,看牛客网的题目描述完全没思路。LeetCode上对于该题解有相应的证明,leetcode1382.
本题目的大体思路是,选取中间节点作为根节点,采用中序遍历构造二叉搜索树,如果数组个数是偶数,选取中间位置右边的数字作为根节点;在LeetCode上可以选取左边位置的数字或者这两种位置随机选择一个。牛客网上只能通过选取中间位置右边的,二者测试样例不同。
class Solution {
public:
/**
*
* @param num int整型vector
* @return TreeNode类
*/
TreeNode* dfs(vector<int>& num,int left,int right)
{
if(left>right) return nullptr;
int mid = (left+right+1)>>1;
TreeNode* temp = new TreeNode(num[mid]);
temp->left=dfs(num,left,mid-1);
temp->right=dfs(num,mid+1,right);
return temp;
}
TreeNode* sortedArrayToBST(vector<int>& num) {
// write code here
int n =num.size();
if(!n) return nullptr;
return dfs(num,0,n-1);
}
};