完全二叉树的权值 Python题解

1. 题目描述

题目链接：“蓝桥杯”练习系统

2. 解题思路

小明要把相同深度的节点的权值加在一起，即将每层的权值加在一起，对比A1,A2+A3,A4+A5+A6+A7，......哪个和最大。

完全二叉树的特点（假设有k层）：

① 1至k-1层的节点是满的，第n层有2^(n-1)个节点

② 第k层可以不满，但节点均集中在左侧

③ 叶子节点只能出现在第k-1层或者第k层

④ 任何一个节点不可以只有右子树而没有左子树

根据满二叉树的性质可知，一棵包含N个节点的完全二叉树有 log(n,2)+1层，第i层有2^(i-1)个节点，则第i层的节点为第2^(i-1)个节点至第(2^i)-1个节点。将它们就和即为每层的权值和。

3. 代码实现

from math import *
n = int(input())
a = [0] + list(map(int,input().split(" ")))
deep = int(log(n,2))+1
res = 0
flag = 1
for i in range(1,deep+1):
    maxnum = sum(a[2**(i-1):2**i])
    if maxnum>res:
        res = maxnum
        flag = i
    elif maxnum==res:
        if flag<i:
            flag=i

print(flag)

 4. 代码解读

a = [0] + list(map(int,input().split(" ")))

因为数组下标是从零开始的，而题目中根的深度为1，因此在输入的N个整数前加上0，保证根的下标为1。

 deep = int(log(n,2))+1

求完全二叉树的深度，以题目为例，N为7，则deep = int(log(n,2))+1= 3

 flag = 1

flag表示第几层

 for i in range(1,deep+1):
    maxnum = sum(a[2**(i-1):2**i])

循环遍历第1层到第deep层，sum(a[2**(i-1):2**i])求每一层的权值和。

 elif maxnum==res:
        if flag<i:
            flag=i

如果有多个深度的权值和同为最大，则选择最小的深度。