华为的转发问题,没有地方测试了,哈哈

# -- coding: utf-8 --
a = input()
N = int(input())
qun = {}  #每一个群里的人
for i in range(N):
    alls = input().split(",")
    qun[i] = alls

people = {} #每一个人在哪些群
for i,j in qun.items():
    for k in j:
        if k not in people:
            people[k] = [i]
        else:
            people[k].append(i)

to_extend = set()  #扩散集合
to_extend.add(a)
visited = set()  #标记集合

while to_extend:
    flag = 0
    people_pop = to_extend.pop()
    if people_pop not in visited:
        visited.add(people_pop)
        quns = people[people_pop] #发布消息的人在哪些群
        for j in quns:
            to_extend = to_extend|(set(qun[j])-visited)

print(len(visited))



#华为##笔试题目##题解#
全部评论
请问华为笔试是可以用python么
点赞 回复
分享
发布于 2019-08-22 10:22

相关推荐

点赞 2 评论
分享
牛客网
牛客企业服务