给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例 :
输入:[3,2,1,6,0,5]
输出:返回下面这棵树的根节点:
6
/ \
3 5
\ /
2 0
\
1
提示:
给定的数组的大小在 [1, 1000] 之间。
递归
package com.loo;
public class MaxBinaryTree {
public static void main(String[] args) {
int[] arr = new int[] {3,2,1,6,0,5};
System.out.println(getMaxBinaryTree(arr).value);
}
public static TreeNode getMaxBinaryTree(int[] arr) {
if (arr == null || arr.length == 0) {
return null;
}
return maxBinaryTree(arr , 0 , arr.length);
}
public static TreeNode maxBinaryTree(int[] arr , int left , int right) {
if (left == right) {
return null;
}
int maxIndex = maxIndex(arr , left , right);
TreeNode root = new TreeNode(arr[maxIndex]);
root.left = maxBinaryTree(arr , left , maxIndex);
root.right = maxBinaryTree(arr , maxIndex + 1 , right);
return root;
}
public static int maxIndex(int[] arr , int left , int right) {
int max_index = left;
for (int i=left;i<right;i++) {
if (arr[i] > arr[max_index]) {
max_index = i;
}
}
return max_index;
}
static class TreeNode {
int value;
TreeNode left;
TreeNode right;
TreeNode(int v) {
value = v;
}
}
}