[ 2023-07-18 🎰 TIL ]

백준 1260번 파이썬

문제

코드

import sys
from collections import deque

input = sys.stdin.readline

n, m, v = map(int, input().split())

graph = [[False] * (n+1) for _ in range(n + 1)]
visited_dfs = [False] * (n+1) # 노드의 방문여부를 판단하는 배열
visited_bfs = [False] * (n+1)
# 노드의 방문여부를 판단하는 배열

for _ in range(m):
  n1, n2 = map(int, input().split())
  graph[n1][n2] = graph[n2][n1] = True
  # 연결된 노드들을 체크
  
def dfs(s):
  visited_dfs[s] = True
  print(s, end= ' ')
  for i in range(1, n+1):
    if (visited_dfs[i] == False and graph[s][i] == True) :
    # 방문한 적이 없고 해당 노드에 값이 존재 할 때 탐색 시작
      dfs(i)
dfs(v)
print()

def bfs(s): # 큐를 사용
  q = deque([s]) # 방문해야 할 곳 추가
  visited_bfs[s] = True
  
  while q: # 방문할 곳이 없으면 종료
    x = q.popleft()
    print(x, end = ' ')
    for i in range(1, n+1):
      if (visited_bfs[i] == False and graph[x][i] == True):
      # 장문한 적이 없고 해당 노드의 값이 존재할 때
        q.append(i) 
        visited_bfs[i] = True

bfs(v)
print()

dfs는 구현하기 어렵지 않지만 bfs가 정말 헷갈린다..ㅠ