问题:
给定一个整数数组 nums ,其中元素已经按 升序 排列,请将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
示例 1:
输入:nums = [-10,-3,0,5,9]
输出:[0,-3,9,-10,null,5]
解释:[0,-10,5,null,-3,null,9] 也将被视为正确答案:
示例 2:
输入:nums = [1,3]
输出:[3,1]
解释:[1,3] 和 [3,1] 都是高度平衡二叉搜索树。
提示:
1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 按 严格递增 顺序排列
解答思路:
当我们想要将一个有序数组转换为二叉搜索树时,我们可以选择数组的中间元素作为根节点,然后将数组分为左右两半。左半部分将构建根节点的左子树,右半部分将构建根节点的右子树。我们可以递归地对左右两半部分进行相同的操作,直到数组为空。
以下是使用 Java 实现的代码:
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int val) {
this.val = val;
}
}
public class Solution {
public static TreeNode sortedArrayToBST(int[] nums) {
// 如果数组为空,则返回 NULL
if (nums == null || nums.length == 0) {
return null;
}
return buildBST(nums, 0, nums.length - 1);
}
// 递归构建二叉搜索树的方法
public static TreeNode buildBST(int[] nums, int start, int end) {
// 如果数组为空,则返回 NULL
if (start > end) {
return null;
}
// 选择中间元素作为根节点
int mid = start + (end - start) / 2;
TreeNode root = new TreeNode(nums[mid]);
// 递归构建根节点的左右子树
root.left = buildBST(nums, start, mid - 1);
root.right = buildBST(nums, mid + 1, end);
return root;
}
public static void main(String[] args) {
int[] nums = { -10, -3, 0, 5, 9 };
TreeNode root = sortedArrayToBST(nums);
// 这里进行适当的树操作或打印
}
}
在上述代码中,我们首先检查数组是否为空。如果数组为空,则直接返回 NULL。然后,我们使用'buildBST'方法递归地构建二叉搜索树。在构建过程中,我们选择数组的中间元素作为根节点,并将数组分为左右两半。对于每一半,我们递归地调用'buildBST'方法构建相应的子树,并将子树连接到根节点的左右子节点上。最后,我们返回构建好的二叉搜索树的根节点。在实际使用中,需要根据具体需求对二叉搜索树进行进一步的操作,例如中序遍历、查找、插入等。
(文章为作者在学习java过程中的一些个人体会总结和借鉴,如有不当、错误的地方,请各位大佬批评指正,定当努力改正,如有侵权请联系作者删帖。)