LeetCode 108 将有序数组转化为二叉搜索树

136 阅读1分钟
给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。

高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
      0
     / \
   -3   9
   /   /
 -10  5
  //这里采用递归的思路,先取l与r的中间值m,根据m构建一个TreeNode,然后通过递归buildTree(nums, l, m - 1)及buildTree(nums, m + 1, r)来设置该node的左右子节点。
  可以选择中间数字作为二叉搜索树的根节点,这样分给左右子树的数字个数相同或只相差 11,可以使得树保持平衡。
class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        return buildTree(nums,0,nums.length-1);
    }
    public TreeNode buildTree(int[]nums,int l ,int r){
        if(l>r){
            return null;
        }
        int m = l+(r-l)/2;
        TreeNode root = new TreeNode(nums[m]);
        root.left = buildTree(nums,l,m-1);
        root.right = buildTree(nums,m+1,r);
        return root;
    }
}

image.png