LeetCode-108.将有序数组转换为二叉搜索树(相关话题:深度优先)

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

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

示例:

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

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

      0
     / \
   -3   9
   /   /
 -10  5

Java代码:

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        if(null == nums || 0 == nums.length)
            return null;
        return dfs(nums, 0, nums.length);
    }
    private TreeNode dfs(int[] nums, int s, int e){
        TreeNode root;
        if(s+1 == e)
            return new TreeNode(nums[s]);
        if(s+2 == e){
            root = new TreeNode(nums[s+1]);
            root.left = dfs(nums, s, s+1);
        }else {
            int p = (e-s)/2, q = (e-s)%2;
            root = new TreeNode(0 == q ? nums[s+p-1] : nums[s+p]);
            root.left = dfs(nums, s, 0 == q ? s+p-1 : s+p);
            root.right = dfs(nums, 0 == q ? s+p : s+p+1, e);
        }

        return root;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_38823568/article/details/82982853