본문 바로가기

코딩테스트/Python

[Python] 백준 1260 - DFS와 BFS

https://www.acmicpc.net/problem/1260

 

1260번: DFS와 BFS

첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사

www.acmicpc.net

 

내 코드

from collections import deque

def dfs(graph,n,visited):
    
    visited[n] = True
    print(n,end=" ")

    for i in graph[n]:
        if visited[i] == False:
            dfs(graph,i,visited)

def bfs(graph,start,visited):
    queue = deque([start])
    visited[start] = True

    while queue:
        v = queue.popleft()
        print(v,end=" ")
        for i in graph[v]:
            if visited[i]==False:
                visited[i] = True
                queue.append(i)


N,M,V = map(int,input().split())

graph = [[] for _ in range(N+1)]
visited = [False for _ in range(N+1)]

for i in range(M):
    l,r = map(int,input().split())
    graph[l].append(r)
    graph[r].append(l)

[g.sort() for g in graph]

dfs(graph,V,visited)
visited = [False for _ in range(N+1)]
print()
bfs(graph,V,visited)

그래프 코드 짜는 법이 기억이 안나서 기초부터 다지고왔다!ㅠㅠ

dfs, bfs 순서대로 구해주었다.

백준에 있는 숏코딩은 해석불가..