Python-交友程式設計

Question

求該題目程式碼

A可以找尋B的朋友，去認識第二層新的朋友，透過這種方法，可以認識一層又一層新的朋友。
求成員X, Y是否有機會可以互相認識。

Answer 1

from collections import deque

def BFS(friends, one, two):
    q = deque()
    q.append(one)
    s = set()
    s.add(one)
    while len(q) > 0:
        friend = q.popleft()
        for i in friends[friend]:
            if i not in s:
                q.append(i)
                s.add(i)
        if friend == two:
            return 'Yes'
    return 'NO'
N, X, Y = map(int, input(">>>").split())

d = {}
for _ in range(N):
    x, y = map(int, input(">>>").split())
    d[x] = d.get(x, []) + [y]
    d[y] = d.get(y, []) + [x]

result = BFS(d, X, Y)
print(result)