본문 바로가기
알고리즘/백준온라인저지 (Python)

[백준온라인저지] 1260 - DFS와 BFS

by 미래문 2021. 8. 24.
반응형


 

 

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

 

1260번: DFS와 BFS

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

www.acmicpc.net

import sys

def dfs(v):
    print(v, end=' ')
    visit[v] = 1

    for i in range(1, n+1):
        if visit[i] == 0 and s[v][i] == 1:
            dfs(i)

def bfs(v):
    queue = [v]
    visit[v] = 0

    while queue:
        v = queue[0]
        print(v, end=' ')
        del queue[0]

        for i in range(1, n+1):
            if visit[i] == 1 and s[v][i] == 1:
                queue.append(i)
                visit[i] = 0

n, m, v = map(int, sys.stdin.readline().split())

s = [ [0] * (n+1) for i in range(n+1)]
visit = [ 0 for i in range(n+1)]

for i in range(m):
    x, y = map(int, sys.stdin.readline().split())

    s[x][y] = 1
    s[y][x] = 1

dfs(v)
print()
bfs(v)
반응형

댓글