树状结构查询

import functools
import collections
import math
from itertools import combinations
from re import match
import copy 
from collections import defaultdict
 
n = int(input())
relation = {}
for i in range(n):
    params = [x for x in input().split(" ")]
    a = params[0]
    b = params[1]
    if b not in relation:
        relation[b] = set([])
    relation[b].add(a)
target = input()
queue = list(relation[target])
result = []
while len(queue) > 0:
    vals = queue[0]
    queue.pop(0)
    result.append(vals)
    if vals in relation:
        for x in list(relation[vals]):
            queue.append(x)
 
result = sorted(result,key=lambda x: x ,reverse=False)
output_str = ""
for ele in result:
    output_str += ele + "\n"
print(output_str)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值