蓝桥杯练习-3.8

蓝桥杯练习-3.8

• G 完全二叉树的权值

Description

给定一棵包含N个节点的完全二叉树，树上每个节点都有一个权值，按从

上到下、从左到右的顺序依次是A1,A2,· · ·AN，如下图所示：

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

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

注：根的深度是1。

Input

第一行包含一个整数N。

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

Output

输出一个整数代表答案。

Sample Input 1

7
1 6 5 4 3 2 1

Sample Output 1

2

Hint

对于所有评测用例，1≤N≤100000，100000≤A_i≤1000001≤N≤100000，100000≤A**i≤100000。

思路

需要了解二叉树的性质

• 在二叉树的第i层上最多有2^(i-1) 个节点 。（i>=1）

• 二叉树中如果深度为k,那么最多有2^k - 1个节点。(k>=1）

• 在完全二叉树中，具有n个节点的完全二叉树的深度为[log2n]+1，其中[log2n]是向下取整。

我们只需要把每一深度的结点的权值加在一起，找到最大的那个即可。

因为测试用例 N ≤ 100000,而2^17 = 131072，所以深度最多k最多为17，而且17层并未填满，也需要判断，是否不是满二叉树的情况。因为最后一层可能并没有填满。

#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
long long ans=-10000000005;//权值总和
int d;//记录最终深度
int a[100005];//输入的每个结点的权值
int main()
{
    int n;//输入多少个结点的权值
    int depth;//计算最大有多深
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
    }
    for (int i = 1; i <= 17; i++)//计算出最大的深度
    {
        if (n >= pow(2,i - 1) && n <= pow(2,i) - 1)
        {
            depth = i;
            break;
        }
    }
    for (int i = 1; i <= depth; i++)//开始循环每一层树了
    {
        long long sum = 0;//每次都要初始化记录权值的变量
        for (int j = pow(2, i - 1); j <= pow(2, i) - 1 && j <= n; j++)//在第i层的开始，第一个是第2^(i-1)-1+1个，最后一个是2^k - 1
            {
                sum += a[j];
            }
            if (sum == ans)//如果有一样大的，取深度最小的那个
            {
                d = min(d, i);
            }
            if (sum > ans)
            {
                ans = sum;
                d = i;
            }
    }
    cout << d << endl;
    return 0;
}