传递悄悄话 (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)