Java完全二叉树的权值

题目描述

给定一棵包含 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)进行计算。