LeetCode算法题108：将有序数组转换为二叉搜索树（Convert Sorted Array to Binary Search Tree）

技术交流可以加：
本人微信：xcg852390212
本人qq：852390212
学习交流qq群1（已满）： 962535112
学习交流qq群2： 780902027

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

LeetCode中文

LeetCode英文

将一个按照升序排列的有序数组，转换为一棵高度平衡二叉搜索树。

本题中，一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例:

给定有序数组: [-10,-3,0,5,9],

一个可能的答案是：[0,-3,9,-10,null,5]，它可以表示下面这个高度平衡二叉搜索树：

      0
     / \
   -3   9
   /   /
 -10  5

解答

考察二叉树的中序建立和平衡二叉树性质，利用平衡二叉搜索树(平衡BST)的性质，平衡BST的中序遍历节点值组成的数组，它的根节点一定是在数组的中间位置。因此对每个子树同样适用这个规律，这就是递归的思想。

对每一个子树组成的数组，有以下几种情况：

1. 数组为空：已经到了空节点，直接返回nullptr；
2. 数组非空：新建一个节点作为当前子树的根节点，它的值取为数组中间位置的值，然后将数组左边部分构成的子数组用来构造左子树，将数组右边部分构成的子数组用来构造右子树。

C++代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* getTree(vector<int>& vec,int l,int r)
    {
        if(l <= r)
        {
            int mid = l + (r - l) / 2;
            TreeNode* node = new TreeNode(vec[mid]);
            node->left = getTree(vec,l,mid - 1);
            node->right = getTree(vec,mid + 1,r);
            return node;
        }
        else
            return nullptr;
    }
    
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        if(nums.size() == 0)
            return nullptr;
        
        return getTree(nums,0,nums.size() - 1);
    }
};

Python代码

# Definition for a binary tree node.
# class TreeNode:

#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def recursion(self,nums: List[int],l: int,r: int) -> TreeNode:
        if l > r:
            return None
        m = (l + r) // 2
        node = TreeNode(nums[m])
        node.left = self.recursion(nums,l,m - 1)
        node.right = self.recursion(nums,m + 1,r)
        return node
    
    def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
        n = len(nums)
        return self.recursion(nums,0,n - 1)