题目描述
给定一棵包含 N 个节点的完全二叉树,树上每个节点都有一个权值,按从上到下、从左到右的顺序依次是 A1,A2,⋯AN,如下图所示:
现在小明要把相同深度的节点的权值加在一起,他想知道哪个深度的节点权值之和最大?如果有多个深度的权值和同为最大,请你输出其中最小的深度。
注:根的深度是 11。
输入格式
第一行包含一个整数 N。
第二行包含 N 个整数 A1,A2,⋯,AN。
输出格式
输出一个整数代表答案。
输入输出样例
输入
7 1 6 5 4 3 2 1
输出
2
说明/提示
对于所有评测用例,1≤N≤10^5,0≤∣Ai∣≤10^5。
import java.util.Scanner;
public class Main{
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int[] arr = new int[n + 1];
for(int i = 1;i < arr.length;i++) {
arr[i] = scanner.nextInt();
}
int high = (int)(Math.log(n) / Math.log(2)) + 1;
int max = 0,max_high = 0;
int sum = 0;
for(int i = 1;i <= high;i++) {
sum = 0;
for(int j = (int)Math.pow(2,i - 1);j < Math.pow(2,i) && j < arr.length;j++) {
sum += arr[j];
}
if(sum > max) {
max = sum;
max_high = i;
}
}
System.out.print(max_high);
}
}
总结:在计算对数时可以使用Math.log(double n)进行计算,方法默认以e为底计算n的对数,因此在此题中想要计算以2为底n的对数可以采用公式:log(n) / log(2)进行计算。