https://www.acmicpc.net/problem/1260
아래는 나의 코드이다. DFS,BFS를 복습하기에 좋은 문제이다. 상하좌우를 검색하는 유형이 아닌 그래프 유형이다.
import sys
from collections import deque
N,M,V = map(int, sys.stdin.readline().split())
myMap = []
for i in range(M):
temp= list(map(int,sys.stdin.readline().split()))
myMap.append([temp[0],temp[1]])
myMap.append([temp[1],temp[0]])
myMap.sort(key= lambda x:x[1])
visited=[False]*(N+1)
DfsResult=[]
BfsResult=[]
def DFS(start):
if visited[start] == True:
return 0;
visited[start]=True
DfsResult.append(start)
for temp in myMap:
if temp[0] == start:
DFS(temp[1])
def BFS(start):
que = deque([start])
while que:
nq = que.popleft()
if visited[nq] == True:
continue
visited[nq]=True
BfsResult.append(nq)
for temp in myMap:
if temp[0] == nq:
que.append(temp[1])
DFS(V)
visited=[False]*(N+1)
BFS(V)
print(*DfsResult)
print(*BfsResult)
'Problem Solving > DFS' 카테고리의 다른 글
BOJ1941 소문난 칠공주 R * (1) | 2024.03.08 |
---|---|
음료수 얼려먹기 (0) | 2022.12.30 |