LeetCode——将有序数组转换为二叉搜索树

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测试结果：