【Python】2019年蓝桥杯省赛真题——完全二叉树的权值

蓝桥杯 2019 省 A&B：完全二叉树的权值

题目描述

给定一棵包含 $N$ 个节点的完全二叉树，树上每个节点都有一个权值，按从上到下、从左到右的顺序依次是 $A_1,A_2,\cdots A_N$，如下图所示：

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

注：根的深度是 $1$。

输入描述

第一行包含一个整数 $N(1\le N\le 10^5)$。

第二行包含 $N$ 个整数 $A_1,A_2,\cdots ,A_N(-10^5\le A_i\le 10^5)$。

输出描述

输出一个整数代表答案。

样例 #1

样例输入 #1

7
1 6 5 4 3 2 1

样例输出 #1

2

运行限制

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

代码实现

n = int(input())
tree = list(map(int, input().split()))
values = []
for i in range(len(bin(n)[2:])):	# 二叉树深度
    values.append(sum(tree[2**i-1: min(2**(i+1)-1, len(tree))]))
print(values.index(max(values)) + 1)