NC_11_SORTEDARRAY_2_BST
package org.example.interview.practice; /** * @author xianzhe.ma * @date 2021/8/18 * 升序数组转二叉平衡树 */ public class NC_11_SORTEDARRAY_2_BST { public TreeNode sortedArrayToBST (int[] nums) { // 判断特殊情况, 数组为空,或数组上没有元素,直接返回 null if (nums == null || nums.length == 0) { return null; } return process(nums, 0, nums.length - 1); } /** * * @param nums 整个的有序数组 * @param left 数组的左边界, 闭区间 * @param right 数组的右边界, 闭区间 * @return nums[left ... right] 这个范围的数组,转成 BST 后的根节点 */ public TreeNode process(int[] nums, int left, int right) { // 左边界 比 右边界 大, 说明数组上没有元素,直接返回 null if (left > right) { return null; } // 如果只有一个元素,就把它当成根节点直接返回 if (left == right) { TreeNode root = new TreeNode(nums[left]); return root; } // nums[left ... right] 这个数组的长度 int len = right - left + 1; // nums[left ... right] 这个数组的中点下标,这个下标里的元素值就是 BST 的根节点的值 int mid = left + len / 2; TreeNode root = new TreeNode(nums[mid]); // 找出根节点的左子树: 继续递归用这个方法,找出左子树上这个局部范围的BST的根节点 root.left = process(nums, left, mid - 1); // 找出根节点的右子树: 继续递归用这个方法,找出右子树上这个局部范围的BST的根节点 root.right = process(nums, mid + 1, right); return root; } public class TreeNode { int val = 0; TreeNode left = null; TreeNode right = null; public TreeNode(int val) { this.val = val; } } }