day 04

简单二叉树的应用

题目描述

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

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

输入

第一行包含一个整数N。 第二行包含N 个整数A1, A2, AN 对于所有评测用例，1<=N<=100000, -100000<=Ai<=100000。

输出

输出一个整数代表答案。

样例输入

1
2

7
1 6 5 4 3 2 1

样例输出

1

2

题目分析

只要计算出每一层的权值，然后求出权重最大所在的深度即可。

每一层可以放的元素个数为　2i−12^{i-1}

题解

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19

n = eval(input())
wight = input().split(' ')
wight = [eval(i) for i in wight]
wight_list = []

i = 0
k = 0
while i < n:
    j = 0
    temp = []
    while j < 2**k:
        if i == n:
            break
        temp.append(wight[i])
        j += 1
        i += 1
    k += 1
    wight_list.append(sum(temp))
print(wight_list.index(max(wight_list))+1)