반응형
https://www.acmicpc.net/problem/11403

bfs나 dfs로 풀 수 있는 문제의 대표적 예시.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import sys | |
from collections import deque | |
n = int(sys.stdin.readline()) | |
maps = [list(map(int, sys.stdin.readline().split())) for _ in range(n)] | |
# y 기준으로 x 탐색 | |
def bfs(start, end, maps): | |
queue = deque() | |
queue.append(start) | |
visited = set() | |
while queue: | |
x = queue.popleft() | |
visited.add(x) | |
next_row = maps[x] | |
for i in range(len(next_row)): | |
if next_row[i] == 1: | |
# 갈 수 있는 곳이 도착지일 경우 | |
if i == end: | |
return 1 | |
# 아직 방문하지 않은 row일 경우 | |
elif i not in visited: | |
queue.append(i) | |
visited.add(i) | |
return 0 | |
answer = [[0 for _ in range(n)] for _ in range(n)] | |
for y in range(n): | |
for x in range(n): | |
answer[y][x] = bfs(y, x, maps) | |
print(*answer[y]) | |
반응형
'프로그래밍 > 코딩테스트 문제풀이' 카테고리의 다른 글
[Python] 백준 1325. 효율적인 해킹 (0) | 2020.02.26 |
---|---|
[Python] 프로그래머스. 2020 카카오 recruit - 외벽 점검 (Level 3) (0) | 2020.02.25 |
[Python] 백준 17406. 배열 돌리기 4 (0) | 2020.02.18 |
[Python] 프로그래머스. 카드 게임 (Level 4) (0) | 2020.02.17 |
[Python] 프로그래머스. 지형 이동 (Level 4) (0) | 2020.02.15 |