完全二叉树的权值

完全二叉树的权值

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

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

输入格式
第一行包含一个整数 N。
第二行包含 N 个整数 A1,A2,⋅⋅⋅AN。

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

数据范围
1≤N≤105,
−105≤Ai≤105
输入样例：
7
1 6 5 4 3 2 1
输出样例：
2

思路：直接暴力比较每一层的总和大小即可，不要忘记前缀和求和时可能爆int，要用long long
代码：

#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=100010;
int n;
int s[N];
LL m[N];
int res=0;
LL maxnum;
int main(){
    cin >> n;
    for(int i=1;i<=n;i++){
        scanf("%d",&s[i]);
    }
    for(int i=1;i<=n;i++){
        m[i]=(LL)s[i]+m[i-1];
    }
    for(int i=1,j=2,cnt=1;i<=n;i=j,j*=2,cnt++){
        LL t;
        if(j<=n)
            t=m[j-1]-m[i-1];
        else
            t=m[n]-m[i-1];
        if(i==1)
        {
            maxnum=t;
            res=cnt;
        }
        else{
            if(t>maxnum)
            {
                maxnum=t;
                res=cnt;
            }
        }
    }
    cout << res<< endl;
    return 0;
}