完全二叉树的权值

原题：

题目描述
给定一棵包含N个节点的完全二叉树，树上每个节点都有一个权值，按从上到下。从左到右的顺序依次是A1，AN，如下图所示:

现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点权值之和最大?如果有多个深度的权值和同为最大，请你输出其中最小的深度。
注:根的深度是1。
输入描述
第一行包含一个整数N(1<N<10^5)。
第二行包含N个整数A1，A…An(-10^5<A≤10^5)。
输出描述
输出一个整数代表答案。

废话不多说，直接上代码：

所有思路都体现在代码的注释里面了。

import java.util.Scanner;

public class day04 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        //输入一个整数N,代表这个完全二叉树有N个节点
        int N = scanner.nextInt();
        //新建一个长度为N的数组存放节点的数据
        int [] N_add = new int[N];
        int k = 0;
        //flag1代表这个完全二叉树的深度
        int flag1 = 1;
        while(k<N){
            N_add[k++] = scanner.nextInt();
            //如果1，3，7，15，也就是到下一深度的节点了，此时要把深度+1；
            if(k==Math.pow(2,flag1)-1){
                flag1++;
            }
        }
        //新建一个数组用来存放每一层节点的权值和，长度自然为完全二叉树的深度
        int[] i_add = new int[flag1];
        flag1 = 0;
        for(k = 0;k<N;k++){
            //如果1，3，7，15，也就是到下一深度的节点了，此时要把深度+1；
            if ((Math.pow(2,flag1) - 1) == k){
                flag1++;
            }
            //我是先处理了深度，但是存放却是前一层的权值和，因此这里的索引应当减一
            i_add[flag1-1] += N_add[k];
        }
        flag1 = 1;
        for(int i:i_add)
            System.out.println("深度为: "+(flag1++)+" 的权值和是: "+i);
        int max_i = i_add[0];
        int result = 1;
        for(int i = 0;i<i_add.length;i++){
            if(max_i<i_add[i]) {
                //更新最大权值和
                max_i = i_add[i];
                //因为数组下标是从0开始的，而对应的深度是从1开始的，所以这里的结果应当加1
                result = i+1;
            }
        }
        System.out.println("深度为: "+result+" 的权值和最大");
    }
}