2019蓝桥杯C++A组——完全二叉树的值

问题描述

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

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

注:根的深度是 1。

【输入】
第一行包含一个整数 N。 第二行包含N个整数A1,A2,··· AN。
对于所有评测用例，1≤ N ≤100000，−100000≤ Ai ≤100000。

【输出】
输出一个整数代表答案。

【输入样例】

7
1 6 5 4 3 2 1

【输出样例】

2

题目解析

1. 做题之前先了解一下什么是完全二叉树
   若设二叉树的深度为h，除第h层外，其它各层(1~h-1)层的结点数都达到最大个数，第h层所有的结点都连续集中在最左边，这就是完全二叉树。
2. 根据题目要求的按从上到下、从左到右的顺序排列，这棵完全二叉树，第一层为第1个数，第2层为其后2个，第3层为其后4个，…，第i层为其后的2^( i−1)个。明白了这个，就一层层把权值加起来就行，从每层中选个最大的就行了。非常简单。

C++代码

#include<bits/stdc++.h>
using namespace std;
int a[101000];
int p(int x) //深度为x的最大标号
{
    int t=1;
    while(x--) t*=2;
    return t;
}
int main()
{
    int n,t=1,i=2,j=2; //i表示遍历所有数，j表示遍历当前层
    cin>>n>>a[1]; //得到结点个数和第一个数的值a[1]
    long long maxn = a[1];
    for(int i=2;i<=n;i++) cin>>a[i];
    while(i<=n)
    {
        long long k=0;
        for(;i<p(j)&&i<=n;i++) k+=a[i]; //k为当前层的权值之和
        if(k>maxn)
        {
            maxn = k; //更新最大层的最大值
            t = j; //更新最大层的层数
        }
        j++;
    }
    cout<<t<<endl; //输出最大权值层的层数
    return 0;
}

样例输入输出

【输入】

【输出】