蓝桥杯-完全二叉树的权值-python解法

题目描述

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

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

注：根的深度是 1。

输入描述

第一行包含一个整数 N（1 \leq N \leq 10^5）N（1≤N≤105）。

第二行包含 N 个整数 A_1, A_2, ··· A_N （−10^5 \leq A_i \leq 10^5）A1​,A2​,⋅⋅⋅AN​（−105≤Ai​≤105）。

输出描述

输出一个整数代表答案。

输入输出样例

示例

输入

7
1 6 5 4 3 2 1

输出

2

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

思路：

题目要求权值最大的节点深度，如果有多个深度的权值和同为最大，输出其中最小的深度。由此可以分成两步，1.求每个深度的权值；2.找到权值最大的节点深度并输出

首先来看第一步，题目给了我们一个串数字，并要求这串数字构成完全二叉树，通过观察，我们发现第1个数在第一层，第2，3个数在第二层，第4，5，6，7个数在第三层，，，，，，以此类推，我们只要找到不同层行的间断点，再将两个间断点之前的所有值相加，就得到了这一层得权值。由上图，观察其A的下标，不难发现层的区间为：2的n次方——2的n+1次方(n=1，2，3.......)，确定了区间，我们把区间内的数相加即可

 第二步，得到每一层的权值后，我们要做的就是找到权值最大的点，输出其深度

p = []
#找到间断点1，2，4，8，16，，，，，
for i in range(0, 33):  
    p.append(2 ** i)    
 
n = int(input())
a = input().split(" ")
 
ans = [0] * 32  #记录每层的和

#遍历所有输入的数
for i in range(0, n):
    #更新每一层的和
    for j in range(0, len(p)):
         #在区间内的数相加求和
        if i + 1 >= p[j] and i + 1 < p[j + 1]:
            ans[j] += int(a[i])

#找到最大值，输出深度
mx, u = -1, -1
for i in range(0, len(ans)):
    if mx < ans[i]:
        mx, u = ans[i], i
print(u + 1)