[蓝桥杯] 完全二叉树的权值

完全二叉树的权值

题目

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

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

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

注：根的深度是 1。

输入格式

第一行包含一个整数 N。

第二行包含 N 个整数 A1,A2,⋅⋅⋅AN。

输出格式

输出一个整数代表答案。

数据范围

$$\begin{gathered} 1\le N\le 10^5 \\ -10^5\le Ai\le 10^5 \end{gathered}$$

输入样例

7
1 6 5 4 3 2 1

输出样例

2

题解

思路

• pow()的精度不够，尽量使用左移操作

#include <iostream>
#include <cmath>

using namespace std;
typedef long long int LL;

const int N = 1e5 + 20, MAX = 0x3f3f3f3f;
LL a[N], n;

int main () {
    cin >> n;
    for (int i = 0; i < n; i ++) cin >> a[i];
    
    LL x = 1, sum = 0, maxv = -MAX, res, cnt = 1;
    for (int i = 0; i < n;) {
        sum = 0;
        while (i < x && i < n) {
            sum += a[i];
            i ++;
        }
        if (maxv < sum) {
            maxv = sum;
            res = cnt;
        }
        cnt ++;
        x = (1 << cnt) - 1;
    }
    cout << res;
    return 0;
}