Acwing---1240. 完全二叉树的权值

1.题目

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

现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点权值之和最大？

如果有多个深度的权值和同为最大，请你输出其中最小的深度。

注：根的深度是 1。

输入格式
第一行包含一个整数 $N$。

第二行包含 N 个整数 A₁,A₂,⋅⋅⋅A_N。

输出格式
输出一个整数代表答案。

数据范围
$1\le N\le 10^5$,
$-10^5\le Ai\le 10^5$

输入样例：

7
1 6 5 4 3 2 1

输出样例：

2

2.基本思想

for(d=1,i=1;i<=n;d++,i*=2){//循环遍历每一层
	sum=0;
	//求出每一层的权值和
	for(j=i;j<=n&& j<i+(1<<(d-1));j++) s+=a[j];
	更新max，depth
}

3.代码实现

import java.util.Scanner;

public class _1240完全二叉树的权值 {
    static Scanner sc = new Scanner(System.in);
    static int a[] = new int[100010];

    public static void main(String[] args) {
        int n = sc.nextInt();
        for (int i = 1; i <= n; i++) a[i] = sc.nextInt();
        long max = Integer.MIN_VALUE;
        int depth=0;
        for (int d = 1, i = 1; i <= n; d++, i *= 2) {//d 深度 层数 i每一层的起点   循环每一层
            long sum = 0;//可能爆 int
            for (int j = i; j < i + (1<<(d-1)) && j <= n; j++) sum += a[j];//每层的权值和
            if (sum > max){
                max=sum;
                depth = d;
            }
        }
        System.out.println(depth);
    }
}