【ACWing】1240. 完全二叉树的权值

题目地址：

https://www.acwing.com/problem/content/1242/

给定一棵包含$N$个节点的完全二叉树，树上每个节点都有一个权值，按从上到下、从左到右的顺序依次是$A_1,A_2,\cdot \cdot \cdot ,A_N$。现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的深度。注：根的深度是$1$。

输入格式：
第一行包含一个整数$N$。第二行包含$N$个整数$A_1,A_2,\cdot \cdot \cdot ,A_N$。

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

数据范围：
$1\le N\le 10^5$
$-10^5\le A_i\le 10^5$

第$k$层的节点的编号是从$2^{k-1}$到$2^k-1$。枚举每一层即可。代码如下：

#include <iostream>
#include <climits>
using namespace std;

const int N = 1e5 + 10;
int n;
long a[N];

int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) scanf("%ld", &a[i]);
    int res = 0;
    long sum = 0, maxs = LONG_MIN;
    for (int k = 1; k <= n; k++) {
        sum = 0;
        int i = 1;
        for (i = 1 << k - 1; i < min((long) n + 1, 1L << k); i++)
            sum += a[i];
    
        if (sum > maxs) {
            res = k;
            maxs = sum;
        }

        if (i == n + 1) break;
    }

    printf("%ld\n", res);

    return 0;
}

时空复杂度$O(N)$。