通过有序数组生成平衡搜索二叉树

最新推荐文章于 2023-04-07 11:12:36 发布

转载最新推荐文章于 2023-04-07 11:12:36 发布 · 1.7k 阅读

每日一练专栏收录该内容

157 篇文章

订阅专栏

本文介绍如何从一个不含重复元素的有序数组构建一棵平衡搜索二叉树的方法。通过选取数组中间元素作为根节点，递归构造左右子树，确保生成的二叉树为平衡树，且中序遍历结果与原数组一致。

【题目】

给定一个有序数组arr，已知其中没有重复值，用这个有序数组生成一棵平衡搜索二叉树，并且该搜索二叉树中序遍历的结果与arr一致。

【代码】

    //通过有序数组生成平衡搜索二叉树
    public static Node generateTree(int[] arr){
        if(arr==null){
            return null;
        }
        return generate(arr,0,arr.length-1);
    }

    private static Node generate(int[] arr, int start, int end) {
        if(start>end){
            return null;
        }
        //有序数组最中间的数生成搜索二叉树的头结点
        //该数左边的数生成左子树
        //该数右边的数生成右子树
        int mid=(start+end)/2;
        Node head=new Node(arr[mid]);
        head.left=generate(arr,0,mid-1);
        head.right=generate(arr,mid+1,end);
        return head;

    }