【问题描述】 给定一棵包含 N 个节点的完全二叉树,树上每个节点都有一个权值,按从 上到下、从左到右的顺序依次是 A1, A2, ··· AN,如下图所示:
现在小明要把相同深度的节点的权值加在一起,他想知道哪个深度的节点 权值之和最大?如果有多个深度的权值和同为最大,请你输出其中最小的深度。 注:根的深度是 1。
【输入格式】
第一行包含一个整数 N。 第二行包含 N 个整数 A1, A2, ··· AN 。
【输出格式】
输出一个整数代表答案。
【样例输入】 7 1 6 5 4 3 2 1
import java.util.Scanner;
public class Main{
public static void main(String[] args) {
Scanner cin= new Scanner(System.in);
int len = cin.nextInt();
int array[]=new int[len];
for(int i=0;i<len;i++) {
array[i]=cin.nextInt();
}
int max=0; //遍历到当前层时,层和的最大值
int result=0;
int high=1; //当前完全二叉树的层数
while(len/2!=0) { //注:巧妙之处,通过完全二叉树的最后一个点的编号不断除以二直到结果为零,除的次数加一即为此完全二叉树的层数
len=len/2;
high++; //计算完后得到完全二叉树的层数
}
for(int i=1;i<=high;i++) { //遍历第1到high层
int m=0; //m是第i层的和
for(int j=(int) Math.pow(2, (i-1));j<(int) Math.pow(2, i);j++) { //得到第j层的和,幂运算:Math.pow(double a,double b)
m=m+array[j-1];
}
if(m>max) { //如果当前层的和大于此层之前的最大值,则发生替换
max=m;
result=i;
}
}
System.out.println(result);
}
}
来源:CSDN
作者:严优秀
链接:https://blog.csdn.net/qq_43463117/article/details/104188126