第十届蓝桥杯 ——完全二叉树的权值

问题描述
给定一棵包含 N 个节点的完全二叉树，树上每个节点都有一个权值，按从上到下、从左到右的顺序依次是 A₁ , A₂ , ··· A_N ，

如下图所示：

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

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

注：根的深度是 1。

输入格式
第一行包含一个整数 N。
第二行包含 N 个整数 A₁ , A₂ , ··· A_N 。

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

样例输入
7
1 6 5 4 3 2 1

样例输出
2

数据范围
1 ≤ N ≤ 10⁵，
−10⁵ ≤ A_i ≤ 10⁵

---

题解
模拟：

解题思路：

1. 通过观察可以发现，每一层的 起点 都是 2 的倍数，终点 为 起点×2 - 1；
2. 因此可以枚举每一层的起点，用 sum 累加每一层的权值；

易错点：

• 由于最多有 10⁵ 个节点，
• 因此最后一层大约有 6 × 10⁴ 个节点，每个节点的权值最大为 10⁵，
• 所以权值之和最大是 6 × 10⁹，超出了 int 的上限；

#include <iostream>
using namespace std;

const int N = 100010;

typedef long long LL;

int n;
int w[N];

int main()
{
    cin >> n;
    for (int i = 1; i <= n; i ++) cin >> w[i];
    
    LL ans = 1, maxv = 0;
    for (int i = 1, d = 1; i <= n; i *= 2, d ++)
    {        
        LL sum = 0;
        for (int j = i; j <= min(2 * i - 1, n); j ++) sum += w[j];
        if(sum > maxv)
        {
            ans = d;
            maxv = sum;
        }
    }
    
    cout << ans << endl;
    return 0;
}

---

蓝桥杯C/C++组省赛历年题