LeetCode【108、将有序数组转换为二叉搜索树】(简单)

265 阅读1分钟

题目描述

这是 LeetCode 上的 14. 最长公共前缀,难度为 简单

关键字: 二叉搜索树有序数组二分查找

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

高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。

示例 1:

示例1 示例1结果

输入:nums = [-10,-3,0,5,9]
输出:[0,-3,9,-10,null,5]
解释:[0,-10,5,null,-3,null,9] 也将被视为正确答案:

示例 2:

示例2

输入:nums = [1,3]
输出:[3,1]
解释:[1,3][3,1] 都是高度平衡二叉搜索树。

提示:

  • 1 <= nums.length <= 104
  • -104 <= nums[i] <= 104
  • nums严格递增 顺序排列

二分遍历

一般看到 有序数组 大多数情况下是使用二分进行操作,有序数组转换二叉搜索树,主要找到中点放入到根节点中,数组左半部分放入左节点、右半部分放入右节点。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        int right = nums.length - 1;
        return generateTree(nums, 0, right);
    }

    public TreeNode generateTree(int[] nums, int left, int right) {
        while(left <= right) {
            // 获取中间值
            int mid = left + (right - left) / 2;
            // 中间值作为根节点
            TreeNode node = new TreeNode(nums[mid]);
            // 左半部分放入左节点
            node.left = generateTree(nums, left, mid - 1);
            // 右半部分放入右节点
            node.right = generateTree(nums, mid + 1, right);
            return node;
        }
        // 没有数据返回 null
        return null;
    }
}

引用

我的 github 仓库已经同步建立,点击访问