2299: 蓝桥杯2019年第十届真题-完全二叉树的权值 Python3

解题思路:

注意事项:
    由于python切片机制比较智能，所以这里不需要考虑最后一层的节点数是否缺少。
参考代码:

import math

length, layer = int(input()), []
tree = list(map(int, input().split()))
depth = int(math.log2(length + 1))
i, j = 0, 0
while i < length:    
    layer.append(sum(tree[i:i + 2 ** j]))
    i = i + 2 ** j
    j += 1

print(layer.index(max(layer)) + 1)