2.25日报

做算法题

题目描述

给

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

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

注：根的深度是 1。

输入描述

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

第二行包含 NN 个整数 A_1, A_2, ··· A_N （−10^5 \leq A_i \leq 10^5）A1​,A2​,⋅⋅⋅AN​（−105≤Ai​≤105）。

输出描述

输出一个整数代表答案。

输入输出样例

示例

输入

7
1 6 5 4 3 2 1

输出

2

 

import java.util.Scanner;
// 1:无需package
// 2: 类名必须Main, 不可修改

public class Main {
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        //在此输入您的代码...
        int N = scan.nextInt();
        int arr[]=new int[N];
        for(int i = 0; i < N; i++){
          arr[i] = scan.nextInt();
        }
        int count = 1;//层数
        int temp = N;
        while(true){//计算二叉树层数
          temp /= 2;
          if(temp == 0){
            break;
          }
          count++;
        }
        int ans = 0;//输出结果层数
        int max = 0;//计算最大值
        int n = 1;
        for(int i = 1; i <=count; i++){
          int sum = 0;
          if(i == count){
            for(int j = (int)Math.pow(2,i-1)-1; j < N; j++){
              sum += arr[j];
            }
            if(sum > max){
              ans = count;
            }
          }else{
            for(int j = n - 1; j < 2*n - 1; j++){
              sum += arr[j];
            }
            if(sum > max){
              max = sum;
              ans = i;
            }
            n *= 2;
          }
        }
        System.out.println(ans);
        scan.close();
    }
}