蓝桥杯 2019 省 A&B：完全二叉树的权值
题目描述
给定一棵包含 N N N 个节点的完全二叉树，树上每个节点都有一个权值，按从上到下、从左到右的顺序依次是 A 1 , A 2 , ⋯ A N A_1,A_2, \cdots A_N A1,A2,⋯AN，如下图所示：
现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的深度。
注：根的深度是 1 1 1。
输入描述
第一行包含一个整数 N ( 1 ≤ N ≤ 1 0 5 ) N (1\le N \le 10^5) N(1≤N≤105)。
第二行包含 N N N 个整数 A 1 , A 2 , ⋯ , A N ( − 1 0 5 ≤ A i ≤ 1 0 5 ) A_1,A_2, \cdots, A_N (-10^5\le A_i \le 10^5) A1,A2,⋯,AN(−105≤Ai≤105)。
输出描述
输出一个整数代表答案。
样例 #1
样例输入 #1
7
1 6 5 4 3 2 1
样例输出 #1
2
运行限制
- 最大运行时间：1s
- 最大运行内存：256M
代码实现
python
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)