【蓝桥杯练习--双指针】1240. 完全二叉树的权值

1240. 完全二叉树的权值

• 完全二叉树：一种特殊的二叉树，除最下一层外，其余每一层都是有两个孩子，最后一层的孩子全部集中在左边。
• 思路：

1. 第n层元素的个数是2ⁿ
2. 遍历每层元素求和，比较得到和最大的情况，返回对应的深度

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

using namespace std;

typedef long long LL;

int main()
{
    int n;
    scanf("%d",&n);
    int arr[n];
    for(int i=1;i<=n;i++)
        scanf("%d",&arr[i]);
    
    LL sum=-1e18;
    LL max=0;
    int k=1, i=1;
    while(i<=n)		//最开始我甚至写了一个cnt函数来计算已经遍历过的层数的元素个数。。。
    {
        int j=i+pow(2,k-1)-1;
        if(j>n) j=n;
        LL tmp=0;
        for(;i<=j;i++)
            tmp+=arr[i];
        //cout << "tmp:" << tmp << endl;
        if(tmp>sum)
        {
            sum=tmp;  //因为数值太大爆int了
            max=k;
        }
        k++;
    }
    
    cout << max;
    return 0;
}

实例代码：
在本质上使用的思想是相同的，但是在具体编写的过程中更加清晰简单。
例如：

1. 直接使用下标来表示已经遍历的元素的个数，因为下标与实际的标号有差异，所以在读取数组时从1开始。
2. 一定要注意数据范围，如果最大值是10⁵，应使用long long类型（因为溢出，调试了半个多小时才意识到。。）
3. n次方可使用位运算实现，如：2ⁿ用(1 << d)表示

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

using namespace std;

typedef long long LL;

const int N = 100010;

int n;
int a[N];

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

收获：

1. 当数据大小为10⁵时，为了防止溢出使用LL
2. 最小值为-1e18
3. 2^d-1用(1 << d - 1)表示
4. 双指针是一种思路，只要是使用到两个下标的都可以说是双指针。

蓝桥杯练习题目汇总