蓝桥杯-2019年-完全二叉树的权值

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

 

 

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

注:根的深度是 1。

输入

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

输出

输出一个整数代表答案。

样例输入

7
1 6 5 4 3 2 1

样例输出

2

题解：首先，整理一下关于满二叉树的知识点

层数	每层节点数	全部节点数
1	2^0	2^1-1
2	2^1	2^2-1
n	2^(n-1)	2^n-1

 

假设这棵树是完全二叉树，也就是除了最后一层对于其他每一层而言，都是满的，同时，完全二叉树和满二叉树的层数计算方法一样，代码中有具体实现，如果这棵树有n个节点，把每个节点放在数组中，数组下标从1开始，那么刚好2^(n-1)是第n层第一个数的下标，2^n-1是第n层最后一个数的下标，但是因为它是完全二叉树，所以最后一层需要单独算

#include<stdio.h>
#include<string.h>
#include<math.h>
#define ll long long
int main()
{
    ll n,maxx=-100010;
    scanf("%lld",&n);
    ll i,j,a[100010];
    for(i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    ll m=n,k=0,flag;
    while(m)//求出来有多少层
    {
        m=m/2;
        k++;
    }
    ll sum;
    for(i=1;i<k;i++)//前k-1层都满足，第k层单独算
    {
        sum=0;
        ll t1=pow(2,i-1),t2=pow(2,i);
        for(j=t1;j<t2;j++)
            sum+=a[j];
        if(sum>maxx)
        {
            maxx=sum;
            flag=i;
        }
    }
    sum=0;
    for(i=pow(2,k-1);i<=n;i++)
        sum+=a[i];
    if(sum>maxx)
    {
        flag=k;
    }
    printf("%lld\n",flag);
    return 0;
}