蓝桥杯——历届真题 完全二叉树的权值-python

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

现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点
　　权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的深度。
　　注：根的深度是 1。
输入格式
　　第一行包含一个整数 N。
　　第二行包含 N 个整数 A1,
　　A2, · · · AN 。
输出格式
　　输出一个整数代表答案。
样例输入
7
1 6 5 4 3 2 1
样例输出
2

import math
N=int(input())  #N个点
tree=list(map(int,input().split()))
max=tree[0]
x=1
deep= int(math.log2(N))+1  #数的深度
temp=1   #每层有多少个结点
temp1=0   #从第几个结点是本层的开头结点
sh=int(math.pow(2,deep-1))  #最后一层开头的结点

for i in range(1,deep+1):
    n=0
    if temp1==sh-1:
        temp=N-sh+1
    for j in range(temp):
        n+=tree[temp1]
        temp1+=1
    if n >max:
        x=i 
        max=n
    temp*=2
print(x)