OD C卷 - 传递悄悄话

传递悄悄话 （100）

• 给定一个二叉树，节点采用顺序存储，如 i=0 表示根节点，2i + 1 表示左子树根，2i + 2 表示右子树根;
• 每个节点站一个人，节点数值表示由父节点到该节点传递消息需要的时间；
• 输出从根节点，将消息传递给所有的人需消耗的时间；
  输入描述：
  0 9 20 -1 -1 15 17 -1 -1 -1 -1 3 2 ；节点的顺序存储；-1表示空节点
  输出描述：
  所有的人都收到消息所消耗的时间 38

示例1
输入：
0 9 20 -1 -1 15 17 -1 -1 -1 -1 3 2
输出：
38

示例2
输入：
0
输出：
0

示例3
输入：
0 9
输出：
9

说明：
还原出二叉树如下

思路：

• 函数递归

total_time = 0


def get_time(root):
    global total_time, n, alist
    if 2*root+1 < n and 2*root+2< n and alist[2*root+1] == -1 and alist[2*root+2] == -1:
        return alist[root]
    elif 2*root+1 >=n and 2*root+2 >=n:
        return alist[root]

    if 2*root+1 <n and alist[2*root+1] != -1:
        total_time = max(total_time, alist[root] + get_time(2*root+1))

    if 2*root+2 < n and alist[2*root+2] != -1:
        total_time = max(total_time, alist[root] + get_time(2*root+2))

    return total_time


alist = list(map(int, input().strip().split()))
n = len(alist)
get_time(0)

print(total_time)