654. 最大二叉树

给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:

  1. 二叉树的根是数组中的最大元素。
  2. 左子树是通过数组中最大值左边部分构造出的最大二叉树。
  3. 右子树是通过数组中最大值右边部分构造出的最大二叉树。

通过给定的数组构建最大二叉树,并且输出这个树的根节点。

Example 1:

输入: [3,2,1,6,0,5]
输入: 返回下面这棵树的根节点:

      6
    /   \
   3     5
    \    / 
     2  0   
       \
        1

注意:

  1. 给定的数组的大小在 [1, 1000] 之间。

解题思路:采用递归建树的方法,每次找到最大的数字,最大的数左边为左子树右边为右子树

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode constructMaximumBinaryTree(int[] nums) {
        TreeNode root=costruct(nums,0,nums.length-1);
        return root;
    }
    public int getMaxIndex(int[] nums,int begin,int end){
        int maxValue=Integer.MIN_VALUE;
        int maxIndex=-1;
        for(int i=begin;i<=end;i++){
            if(nums[i]>maxValue){
                maxValue=nums[i];
                maxIndex=i;
            }
        }
        return maxIndex;
    }
    public TreeNode costruct(int[] nums,int begin,int end){
        int index=getMaxIndex(nums,begin,end);
        if(index==-1)return null;
        TreeNode root=new TreeNode(nums[index]);
        root.left=costruct(nums,begin,index-1);
        root.right=costruct(nums,index+1,end);
        return root;
    }
}

 

你可能感兴趣的:(LeetCode,树的处理)