【蓝桥杯】完全二叉树的权值--第十届蓝桥杯省赛C++A/B组

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

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

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

注：根的深度是 1。

输入格式
第一行包含一个整数 N。

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

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

数据范围

1≤N≤105,
−105≤Ai≤105

输入样例：

7
1 6 5 4 3 2 1

输出样例：

2

首先得知道完全二叉树是个什么玩意：

定义啥的我是记不清楚了，但是大白话来说，就是这个二叉树是从左到右开始一个一个的画下来的，没有隔着蹦着都的，最后一层可以不满，但是上面肯定都是满的，大概就这个意思，上图就是一个完全二叉树：

百度百科：
完全二叉树是效率很高的数据结构，完全二叉树是由满二叉树而引出来的。对于深度为K的，有n个结点的二叉树，当且仅当其每一个结点都与深度为K的满二叉树中编号从1至n的结点一一对应时称之为完全二叉树。

然后这道题就好说了，每一层的节点数是2^d-1个（d是从1开始的）
每次遍历每一层所有的数相加，然后遍历完之后进行一个判断，记录最大值和层数最后输出就好了；
代码如下：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;
typedef long long LL;

int a[N];
int n;

int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= n; i ++ )   scanf("%d", &a[i]);
    
    int depth = 0;
    LL max = -1e18;
    
    for(int d = 1, i = 1; i <= n; i *= 2, d ++ )
    {
        LL sum = 0;
        
        for(int j = i; j < i + (1 << d - 1) && j < n; j ++ )
            sum += a[j];
            
        if(sum > max)
        {
            max = sum;
            depth = d;
        }
    }
    
    printf("%d", depth);
    
    return 0;
}