From 5892ceac18f32cc4ca713d9a746bb7679fcf9391 Mon Sep 17 00:00:00 2001 From: park jiyeong Date: Wed, 7 Aug 2024 17:25:10 +0900 Subject: [PATCH] 39 --- JYPARK/31to40/39.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 JYPARK/31to40/39.py diff --git a/JYPARK/31to40/39.py b/JYPARK/31to40/39.py new file mode 100644 index 0000000..5a459a1 --- /dev/null +++ b/JYPARK/31to40/39.py @@ -0,0 +1,26 @@ +from collections import defaultdict, deque + +def solution(graph, start): + tree = defaultdict(list) + visited = set() + answer = [] + for i, j in graph: + tree[i].append(j) + return bfs(tree, start, visited, answer) + +def bfs(tree, node, visited, answer): + q = deque() + q.append(node) + + while q: + node = q.popleft() + if node not in visited: + visited.add(node) + answer.append(node) + q.extend(tree[node]) + return answer + +#graph = [(1, 2), (1, 3), (2, 4), (2, 5),(3, 6),(3, 7),(4, 8), (5, 8),(6, 9),(7, 9)] +graph = [(0,1), (1, 2), (2, 3), (3, 4),(4, 5),(5, 0)] +start = 1 +print(solution(graph, start)) \ No newline at end of file