leetcode 108. 将有序数组转换为二叉搜索树
给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
解题思路:
(1)二叉搜索树的中序遍历是升序序列,题目给定的数组是按照升序排序的有序数组,因此可以确保数组是二叉搜索树的中序遍历序列。
(2)二叉搜索树的中序遍历,并不能可以唯一地确定二叉搜索树,题目中虽然增加了一个限制条件,即要求二叉搜索树的高度平衡,但仍然是不能唯一地确定二叉搜索树。(因为选择根节点的方式不同,产生出来的二叉树也是不同的)
(3)在这里有三种选择根节点的方式:(奇数个时直接取中间,偶数个节点时有三种)
(a)中间位置左边的数字作为根节点
(b)选择中间位置右边的数字作为根节点
(c)选择任意一个中间位置数字作为根节点
在这里我选择了方案一:中间位置左边的数字作为根节点
实际上是以中序遍历的顺序为二叉树一一赋值(数组左右两侧的元素)
class Solution
{
public:
TreeNode* sortedArrayToBST(vector<int>& nums)
{
return begin(nums,0,nums.size()-1);
}
TreeNode *begin(vector<int> &nums,int start,int end)
{
if(start>end) return NULL; //当仅剩一个节点(叶子节点)时,会出现start=end的情况
int mid=(start+end)/2; //默认左侧的节点为(子树)根节点
TreeNode *root=new TreeNode(nums[mid]); //以中间节点作为(子树)根节点
root->left=begin(nums,start,mid-1); //中序遍历产生左子树
root->right=begin(nums,mid+1,end); //中序遍历产生右子树
return root; //不断地返回每一个产生的节点 最后会返回给二叉树的最终根节点
}
};