题目:
给定一个不重复的整数数组 nums
。 最大二叉树 可以用下面的算法从 nums
递归地构建:
- 创建一个根节点,其值为
nums
中的最大值。 - 递归地在最大值 左边 的 子数组前缀上 构建左子树。
- 递归地在最大值 右边 的 子数组后缀上 构建右子树。
返回 nums
构建的 最大二叉树 。
思路:递归法。
第一步:确定参数与返回值。参数为nums数组,要遍历区域的起始下标beginIndex,终止下标endIndex,区间左闭右开,返回值为树。
第二步:确定终止条件。当endIndex-beginIndex<1时,说明数组里无元素,返回null
第三步:确定单层递归逻辑。当endIndex-beginIndex=1,说明数组里只有一个元素,为该元素创建一个新节点;当endIndex-beginIndex>1时,遍历数组,找到最大元素的值与下标,创建节点root存储该元素,递归root.left与root.right。
代码:
public TreeNode constructMaximumBinaryTree(int[] nums) {
return construct(nums,0,nums.length);
}
public TreeNode construct(int[] nums,int beginIndex,int endIndex){
if(endIndex-beginIndex<1) return null;//没有元素了
if(endIndex-beginIndex==1) return new TreeNode(nums[beginIndex]);//只有一个元素
int maxIndex=beginIndex;//最大值下标
int maxVal=nums[beginIndex];//最大值
for(int i=beginIndex+1;i<endIndex;i++){
if(nums[i]>maxVal){
maxIndex=i;
maxVal=nums[i];
}
}//寻找最大值及下标
TreeNode root=new TreeNode(maxVal);
root.left=construct(nums,beginIndex,maxIndex);
root.right=construct(nums,maxIndex+1,endIndex);
return root;
}