目录
给你一个整数数组
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,null,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 { // 「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树 // 即将数组看作中序遍历[左根右],那么数组中间就看做根节点, // 递归处理:根据中序遍历确定根节点,然后递归确定左子树,右子树 public TreeNode sortedArrayToBST(int[] nums) { // 给出数组,本次需确定树的数组范围,初始为0——数组结束 return fun(nums,0,nums.length-1); } TreeNode fun(int[] nums,int beg,int end){ // 递归终止条件:该部分数组内容处理完了 if(beg>end) return null; // 按照中间(靠右)为根节点 int mid=(end-beg+1)/2+beg; // 确定当前范围根节点 TreeNode root=new TreeNode(nums[mid]); // 递归 root.left=fun(nums,beg,mid-1); root.right=fun(nums,mid+1,end); // 返回当前树 return root; } }
运行结果: