蓝桥杯 完全二叉树的权值

题目描述

给定一棵包含 N 个节点的完全二叉树，树上每个节点都有一个权值，按从 上到下、从左到右的顺序依次是 A1, A2, ··· AN​，如下图所示：

现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点 权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的深度。

注：根的深度是 1。

输入描述

第一行包含一个整数 N（1≤N≤10^5）。

第二行包含 N 个整数 A1, A2, ··· AN ​（−10^5≤Ai​≤10^5）。

输出描述

输出一个整数代表答案。

输入输出样例

示例

输入

7
1 6 5 4 3 2 1

输出

2

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

解题思路

       这题只要把握到完全二叉树的性质就很容易解题。

       二叉树每层的叶子数：2^(i-1)

       二叉树到第i层的叶子总数：2^i -1

import java.util.*;

public class Main {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
	    Scanner scan=new Scanner(System.in);
	    int n=scan.nextInt();
	    int tree[]=new int[n+1];
	    for(int i=1;i<=n;i++) {
	    	tree[i]=scan.nextInt();
	    }
	    scan.close();
//	    int num=0;//树的层数
//	    while(n>0) {
//	    	n/=2;
//	    	num+=1;//求树的层数
//	    }
	    int sum=0;
	    int max=-100001;
	    int layer=0;
	    
	    for(int i=1,num=1;num<=n;i++,num*=2) {
	    	sum=0;//测试用例有负数，不可以以0为评判点，而是一个比测试用例还小的负数
	    	for(int j=num;j<=(1<<i)-1&&j<=n;j++) {//忘记考虑最后一层不满的情况了
	    		sum+=tree[j];//求第i层的和
	    	}
	    	if(sum>max) {
	    		max=sum;
	    		layer=i;
	    	}
	    }
	    System.out.println(layer);
	}	
}

收获

       一开始我在eclipse上测试用例是没有问题的，但是在oj上得到“无效返回”的提示。主要因为以下原因：

     （1）我在忘记考虑最后一层不满这种情况了。最后加上 j<=n 解决。

     （2）我的 max 设置为0，但我没有考虑到测试数据的范围是 −10^5≤Ai​≤10^5 。负数漏考虑了。所以通不过为负数的测试用例。我通过设置max= -100001解决。其实还可以 max=Integer.MIN_VALUE。（ Integer.MIN_VALUE为int的最小值 -2 147 483 648）

       其他：

     （1）计算 2^i ：可以用 Math.pair(2,i) ,也可以用位运算，左移也相当于乘2，左移i位相当于2^i，即 1<<i 。

     （2）原本我单独用一个while循环来计算层数，但是看了别人的博客后发现可以将其一起放到循环内，随着层数的增加循环乘2。