有序数组转为平衡二叉树

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

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

示例 1:

输入:nums = [-10,-3,0,5,9]
输出:[0,-3,9,-10,null,5]
解释:[0,-10,5,null,-3,null,9] 也将被视为正确答案:
在这里插入图片描述

输入:nums = [1,3]
输出:[3,1]
解释:[1,3] 和 [3,1] 都是高度平衡二叉搜索树。
在这里插入图片描述

有序数组转为平衡二叉树

问题分析:
若不规定平衡,则可以很多种选择,可以任一节点作为根节点。

有了平衡的限制,数组到二叉树的映射选择就少了很多,要以中间节点作为根节点,减少左右子树的深度差。

根节点左和右子树也是一颗平衡二叉树,因此同样的去在左右子树选择根节点,整个问题的关键就成了根节点的选择。

数组元素为奇数个数时,根节点的选择都是同样的,数组元素为偶数个数时,根节点的下标索引有不同选择:

//向下取整
mid = (start+end)/2;
//向上取整
mid = 1+(start+end)/2;

Java代码实现

/**
 * 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;
      }
      public static void main(String args[]){
          int [ ]nums = {-10,-3,0,5,9};
          Solution solution = new Solution();
          TreeNode root = solution.sortedArrayToBST(nums);
          solution.midOrder(root);
      } 
  }
class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        TreeNode root = buildBstHelper(nums,0,nums.length-1);
        return root;
    }

    public TreeNode buildBstHelper(int[] nums,int start,int end){
        //递归终止条件
        if(start>end){
            return null;
        }
        //可能会溢出
        int mid = (start+end)/2;
        TreeNode root = new TreeNode(nums[mid]);
        root.left = buildBstHelper(nums,start,mid-1);
        root.right = buildBstHelper(nums,mid+1,end);
        return root;
    }

    public void midOrder(TreeNode root){
        if(root==null){
            return;
        }
        midOrder(root.left);
        System.out.println(root.val);
        midOrder(root.right);
    }
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值