当前位置:   article > 正文

2024年华为OD机试真题-传递悄悄话-Python-OD统一考试(C卷)_给订一个二叉树,每个节点上站着一个人, 节点数字标识父节点到该节点传递悄悄话

给订一个二叉树,每个节点上站着一个人, 节点数字标识父节点到该节点传递悄悄话

题目描述:

给定一个二叉树,每个节点上站着一个人,节点数字表示父节点到该节点传递悄悄话需要花费的时间。
初始时,根节点所在位置的人有一个悄悄话想要传递给其他人,求二叉树所有节点上的人都接收到悄悄话花费的时间。

输入描述:

给定二叉树 

0 9 20 -1 -1 15 7 -1 -1 -1 -1 3 2

注:-1表示空节点

输出描述:

返回所有节点都接收到悄悄话花费的时间38

补充说明:

示例1

输入:

0 9 20 -1 -1 15 7 -1 -1 -1 -1 3 2
输出:

38
说明:

  1. from collections import deque
  2. queue = deque()
  3. result = 0
  4. def check(index, nums, father):
  5. global result
  6. if index < len(nums) and nums[index] != -1:
  7. nums[index] += nums[father]
  8. queue.append(index)
  9. if nums[index] > result:
  10. result = nums[index]
  11. if __name__ == "__main__":
  12. input_str = input()
  13. tmp2 = input_str.split(" ")
  14. nums = [int(num) for num in tmp2]
  15. queue.append(0)
  16. while queue:
  17. father = queue.popleft()
  18. check(2 * father + 1, nums, father)
  19. check(2 * father + 2, nums, father)
  20. print(result)

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/人工智能uu/article/detail/775228
推荐阅读
相关标签
  

闽ICP备14008679号