【蓝桥杯试题 历届真题 完全二叉树的权值】

问题描述

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



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

输入格式

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

输出格式

　　输出一个整数代表答案。

样例输入

7
1 6 5 4 3 2 1

样例输出

2

评测用例规模与约定

　　对于所有评测用例，1 ≤ N
　　≤ 100000，−100000 ≤ Ai ≤ 100000。

#include<iostream>
#include<math.h>

using namespace std;

const int N = 100010;

long long tree[N], maxsize = -0x3f3f3f;//初始为负无穷大
int main()
{
    int n, res;
    cin >> n;

    //输入所有节点
    for (int i = 1; i <= n; i++)
    {
        cin >> tree[i];
    }

    for (int i = 1; i <= n; i *= 2)
    {
        long long s = 0;
        //各层子节点相加
        for (int j = i; j <= i * 2 - 1 && j <= n; j++)
        {
            s += tree[j];
        }
        //记录相加做多的层数
        if (s > maxsize)
        {
            maxsize = s;
            res = (int)log2(i) + 1;
        }
    }
    cout << res;
    return 0;
}