蓝桥杯真题 完全二叉树的权值 c++代码实现 给定一棵包含 N 个节点的完全二叉树，树上每个节点都有一个权值，按从上到下、从左到右的顺序依次是 A1, A2, ··· AN

题目描述

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

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

输入描述

第一行包含一个整数 N（1≤N≤10⁵）。

第二行包含 N 个整数A₁​ ,A₂​ ,⋅⋅⋅A_N​（−10⁵ ≤A_i​ ≤10⁵ ）。

输出描述

输出一个整数代表答案。

输入输出样例

示例

输入
7 1 6 5 4 3 2 1

输出
2

运行限制

最大运行时间：1s
最大运行内存: 256M

代码

#include <iostream>
#include <cmath>
using namespace std;
int main()
{
    int N, dep, x, d;
    cin >> N;
    long long num = 0;
    dep = log2(N + 1);//二叉树的深度
    d = 1;//默认为1
    long long arr[10005] = { 0 };//初始化数组
    for (int i = 1; i <= N; i++)
    {
        cin >> x;
        int id = ceil(log2(i + 1));//计算当前的数据所在的深度
        arr[id] += x;//把对应的深度的总和存起来
    }
    for (int i = 1; i <= dep + 1; i++)
    {
        if (num < arr[i])//比较得出最大和的深度
        {
            num = arr[i];
            d = i;
        }
    }
    cout << d;
    return 0;
}

思路

这道题目就是对完全二叉树的层序遍历，遍历每层的二叉树计算出每层的总和，然后找出最大的和。用到了log2()函数，计算以2为底的对数，还有ceil()向上取整函数，头文件是#include <cmath>。完全二叉树的深度h与节点数N的关系为h=log2(N+1)。