2024. 1华为od机试C卷【传递悄悄话】Python

题目

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

输入描述

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

原题

112. 路径总和 - 力扣（LeetCode）

代码

from collections import deque


class TreeNode:
    def __init__(self, val):
        self.val = val
        self.lef = None
        self.rig = None


def list2tree(nums):
    def level(index):
        if index >= len(nums) or nums[index] == -1:
            return None

        root = TreeNode(nums[index])
        root.lef = level(2 * index + 1)
        root.rig = level(2 * index + 2)
        return root

    return level(0)



class Solution:
    def max_time(self, nums):
        root = list2tree(nums)
        if not root:
            return -1
        que_node = deque([root])
        que_val = deque([root.val])
        max_value = 0
        while que_node:
            now = que_node.popleft()
            temp = que_val.popleft()
            if not now.lef and not now.rig:
                if temp > max_value:
                    max_value = max(temp, max_value)
                continue
            if now.lef:
                que_node.append(now.lef)
                que_val.append(now.lef.val + temp)
            if now.rig:
                que_node.append(now.rig)
                que_val.append(now.rig.val + temp)
        return max_value


def main():
    lst = list(map(int, input().split(' ')))
    solution = Solution()
    print(solution.max_time(lst))


if __name__ == "__main__":
    main()