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

Derrickhang

已于 2022-03-23 22:29:44 修改

阅读量341

点赞数

文章标签：算法 c++ 数据结构蓝桥杯

于 2022-03-23 22:27:43 首次发布

本文链接：https://blog.csdn.net/Derrickhang/article/details/123697838

版权

AcWing 1240. 完全二叉树的权值

一、题目链接
二、题目分析
- （一）算法标签
- （二）解题思路
三、AC代码
四、其它题解

一、题目链接

AcWing 1240. 完全二叉树的权值

二、题目分析

（一）算法标签

双指针二叉树

（二）解题思路

含n个结点的满二叉树的深度为 $\lfloor log_{2}{n} \rfloor + 1$ 或 $\lceil log_{2}{(n+1)} \rceil$

cout << (int)(log(1e5) / log(2)) + 1 << endl;
cout << -pow(2, 16) * 1e5;
// output:
17
-6.5536e+09

含有 $10^5$ 个结点对应的满二叉树的深度为17, 每一层最多有 $2^{n-1}$ 个结点，因此结点之和最小为 $2^{n - 1} \times {(-10^{5})} = -6.5536 \times 10^9$ 比 INT_MIN (2147483647)还要小，所以要sum为long long类型
思路：
遍历每一层，计算每一层的和，如果大于当前的sum，则更新maxSum，同时更新答案为当前层数下标

三、AC代码

解法一:

#include <iostream>
#include <cstring>
#include <algorithm>
#include <climits>
#include <cmath>

using namespace std;

typedef long long LL;

const int N = 1e5 + 10;

int a[N];

int n;

int main()
{
    cin >> n;
    for (int i = 1; i <= n; i ++ ) scanf("%d", &a[i]);
    
    LL maxSum = -1e18;
    int res;
    for (int i = 1; i <= (int)(log(n) / log(2)) + 1; i ++ ) // n 个 结点的二叉树的深度为log(n + 1)/log(2) + 1 (向上取整)
    {
        LL sum = 0;
        for (int j = pow(2, i - 1); j <= pow(2, i) - 1 && j <= n; j ++ )    // 题目数据不是满二叉树，因此要判断j是否小于n
            sum += a[j];
        if (sum > maxSum)
        {
            maxSum = sum;
            res = i;
        }
    }
    cout << res << endl;
    // cout << log(1e5) / log(2) + 1 << endl;
    // cout << -pow(2, 17) * 1e5;
    return 0;
}