目录删除

标题:目录删除 | 时间限制:1秒 | 内存限制:262144K | 语言限制:不限
某文件系统中有N个目录,每个目录都一个独一无二的ID。每个目录只有一个父目录,但每个父目录下可以有零个或者多个子目录,目录结构呈树状结构。
假设,根目录的ID为0,且根目录没有父目录,其他所有目录的ID用唯一的正整数表示,并统一编号。

rm_list_all = []
def find_rm(rm_pv, tree_):
    rm_list_all.append(rm_pv)
    if tree_.get(rm_pv):
        for i in tree_.get(rm_pv):
            find_rm(i, tree_)


if __name__ == '__main__':
    lines = int(input().strip())
    tree = {}
    for i in range(lines):
        s, p = [int(i) for i in input().strip().split()]
        if tree.get(p) is None:
            tree[p] = [s]
        else:
            tree[p].append(s)
    rm_p = int(input().strip())
    find_rm(rm_p,tree)
    res = []

    for k,v in tree.items():
        if k not in rm_list_all and k != 0:
            res.append(k)
        for i in v:
            if i not in rm_list_all and i != 0:
                res.append(i)
    res = list(set(res))
    res.sort()
    print(' '.join([str(i) for i in res]))


全部评论

相关推荐

11-29 00:55
门头沟学院
区域赛银,邀请赛金,打算十二月打下Java基础、背点八股、写个外卖后去投福建小厂的寒假实习,简历应该怎么写呢?以及福州/和厦门有推荐的小厂吗?
牛客53210502...:简历一页:把区域银,邀请赛金标粗,其他的奖除非凑一页否则没有必要写。或者多页:每个站一行这样都列出来。项目经历看看牛客其他人是怎么写的,写的不好呢。简历打磨好按部就班没问题的
点赞 评论 收藏
分享
八极星:有什么不能问的,(/_\),这又不是多珍贵的机会,你有什么可失去的
点赞 评论 收藏
分享
LZStarV:冲就好了,就算真的是字节也冲,面评脏了大不了等三四个月就淡了,而且等到那个时候实力进步了选择还多,何必拘泥于字节
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务