LeetCode——将有序数组转换为二叉搜索树
题目描述:
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
一个二叉搜索树的中序遍历就是一个从小到大的有序数组。
因此我们可以把给定的数组看成是一个二叉搜索树的中序遍历。
我们可以选择数组中间的数字作为二叉搜索树的根节点,将左边的子数组作为左子树,右边的子数组作为右子树,这样左右子树的高度差就会小于等于1,满足二叉平衡树的条件。
因为一个中序遍历写出来的二叉搜索树可能不唯一,所有我们需要分情况讨论一下:
当数组中元素个数为奇数,则中间的数字是(left + right) / 2 ,中间的数只有一个。因此根节点的选择唯一。
当数组中元素个数为奇数,则中间的数字可以是中间位置左边的数,也可以是中间位置右边的数,也就是根节点有两种选择,即(left + right)/ 2和(left + right + 1)/ 2, 这里我们选择前一种方式。
在确定好根节点后,根节点的左右子树也为一颗二叉搜索树,因此可以使用递归的方法进行调用。当left大于right时根节点为空,此为递归结束的条件。
这里提供Java和Python的代码。
Java代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
return solution(nums, 0, nums.length - 1);
}
public TreeNode solution(int nums[], int left, int right){
if (left > right){
return null;
}
int mid = (left + right) / 2;
TreeNode root = new TreeNode(nums[mid]);
root.left = solution(nums, left, mid - 1);
root.right = solution(nums, mid + 1, right);
return root;
}
}
LeetCode测试结果:
Python代码:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def sortedArrayToBST(self, nums):
"""
:type nums: List[int]
:rtype: TreeNode
"""
return self.solution(nums, 0, len(nums) - 1)
def solution(self, nums, left, right):
if left > right:
return None
mid = (left + right) / 2
root = TreeNode(nums[mid])
root.left = self.solution(nums, left, mid - 1)
root.right = self.solution(nums, mid + 1, right)
return root
LeetCode测试结果: