华为OD机试 - 空间占用计算(Python)

import sysdef solve_prefix():    lines = sys.stdin.read().strip().splitlines()    if not lines:        return    target = lines[0].strip()    # 第二行：文件列表，可能为空    if len(lines) > 1:        file_paths = lines[1].strip().split()    else:        file_paths = []    # 第三行：大小列表    if len(lines) > 2:        sizes = list(map(int, lines[2].strip().split()))    else:        sizes = []    # 构建路径->大小映射    size_map = {}    for p, s in zip(file_paths, sizes):        size_map[p] = s    # 确保目标目录路径末尾不带斜杠    target = target.rstrip('/')    # 存储一级子项的总大小    sub_size = {}    for path, fsize in size_map.items():        # 检查是否在目标目录下（包括目录本身？但目录本身不是文件，所以必须深度大于）        # 如果目标目录是根目录？处理前缀匹配：要么路径等于目标目录（作为文件？但目标目录是目录，不会直接是文件），        # 要么路径以 target + '/' 开头        if path == target:            # 目标目录本身作为文件？实际上目标目录是目录，不可能有同名文件，忽略            continue        if path.startswith(target + '/'):            # 提取相对路径            rel = path[len(target)+1:]  # 去掉target和后面的/            # 取第一级子项名（第一个/之前的部分）            if '/' in rel:                sub_name = rel.split('/', 1)[0]            else:                sub_name = rel            sub_size[sub_name] = sub_size.get(sub_name, 0) + fsize    if not sub_size:        print("")        return    max_size = max(sub_size.values())    result = [name for name, size in sub_size.items() if size == max_size]    result.sort()    print(' '.join(result))if __name__ == "__main__":    solve_prefix()

import sysclass TreeNode:    def __init__(self, name):        self.name = name        self.children = {}  # name -> TreeNode        self.total_size = 0  # 子树总大小（包括所有后代文件）def solve_tree():    lines = sys.stdin.read().strip().splitlines()    if not lines:        return    target = lines[0].strip()    if len(lines) > 1:        file_paths = lines[1].strip().split()    else:        file_paths = []    if len(lines) > 2:        sizes = list(map(int, lines[2].strip().split()))    else:        sizes = []    # 构建根节点（虚拟根）    root = TreeNode("")    # 插入每个文件    for path, fsize in zip(file_paths, sizes):        parts = path.split('/')        # 跳过空的部分（如第一个/导致空字符串）        parts = [p for p in parts if p]        cur = root        for i, part in enumerate(parts):            if part not in cur.children:                cur.children[part] = TreeNode(part)            cur = cur.children[part]        # 文件节点累加大小（叶子节点）        cur.total_size += fsize        # 向上更新祖先节点总大小        # 这里我们在插入后统一计算，也可以在插入时递归更新，但为避免重复，可以直接在最后递归计算        # 简便方法：插入时向上更新        # 但为了简单，我们之后对整棵树后序遍历计算    # 后序遍历计算每个节点的总大小（包括所有后代文件）    def postorder(node):        total = node.total_size  # 如果是文件节点，已有自己的大小；目录节点初始为0        for child in node.children.values():            total += postorder(child)        node.total_size = total        return total    postorder(root)    # 找到目标目录对应的节点    target_parts = [p for p in target.split('/') if p]    cur = root    for part in target_parts:        if part in cur.children:            cur = cur.children[part]        else:            # 目标目录不存在            print("")            return    # cur 为目标目录节点，其直接子节点即一级子项    if not cur.children:        print("")        return    max_size = max(child.total_size for child in cur.children.values())    result = [name for name, child in cur.children.items() if child.total_size == max_size]    result.sort()    print(' '.join(result))if __name__ == "__main__":    solve_tree()