[Silver I] 현이의 로봇 청소기 - 30106 (python)

문제 링크

 

 

분류

너비 우선 탐색, 플러드 필, 그래프 이론, 그래프 탐색

 

문제 설명

현이의 방은 매우 지저분하다! 현이는 선물로 로봇 청소기를 받았고, 귀찮은 청소를 맡기기로 했다.

로봇 청소기는 현이의 방을 1×1 크기의 정사각형으로 나누어져 있는 𝑁×𝑀 크기의 직사각형으로 인식한다.

로봇 청소기는 한 영역을 청소하고 나서 상, 하, 좌, 우로 인접한 영역 중 하나로 이동한다. 현이의 방은 오래되어 마루가 울퉁불퉁하고, 쓰레기도 눌러붙어 각 영역마다 높이가 조금씩 다르다. 로봇 청소기는 높이 차이가 𝐾 초과인 두 영역 사이를 이동하면 고장이 날 수 있기 때문에, 높이 차이가 𝐾 이하인 영역 사이에서만 이동한다.

현이는 외출하는 동안 로봇 청소기를 작동시키고 집에 돌아왔지만, 청소가 되지 않은 곳도 있는 것을 보고 실망했다.

결국 현이는 로봇 청소기의 위치를 직접 옮겨주기로 했다. 현이가 로봇 청소기를 직접 옮길 때는 아무 영역으로나 옮길 수 있다. 로봇 청소기를 켠 상태로 옮기면 위험하니 로봇 청소기를 끄고 옮긴 뒤에 다시 작동시킬 수 있다.

현이가 로봇 청소기를 최소 몇 번 작동시켜야 방의 모든 영역을 청소할 수 있을지 구해보자!

입력

첫 번째 줄에 자연수 𝑁, 𝑀, 𝐾가 공백으로 구분되어 주어진다.

두 번째 줄부터 𝑁+1번째 줄까지 각 영역의 높이를 나타내는 정수 𝑀개가 공백으로 구분되어 주어진다.

출력

방의 모든 영역을 청소하기 위해 로봇 청소기를 작동시켜야 하는 최소 횟수를 출력한다.

 

 

코드

import sys
from collections import deque
input = sys.stdin.readline


def bfs(x, y):
    # 현재 위치를 방문했음을 표시
    vis[x][y] = 1

    q = deque([(x, y)])
    while q:
        r, c = q.popleft()

        # 상하좌우 네 방향에 대해 검사
        for i in [(-1, 0), (1, 0), (0, -1), (0, 1)]:
            nr = r + i[0]
            nc = c + i[1]

            # 새로운 위치가 유효한 범위 내에 있고, 방문하지 않았으며, 높이 차이가 k 이하인 경우
            if 0 <= nr < n and 0 <= nc < m and not vis[nr][nc]:
                if abs(room[r][c] - room[nr][nc]) <= k:
                    vis[nr][nc] = 1
                    q.append((nr, nc))


n, m, k = map(int, input().split())
room = [list(map(int, input().split())) for _ in range(n)]
vis = [[0]*m for _ in range(n)]
ans = 0

for i in range(n):
    for j in range(m):
        if not vis[i][j]:
            bfs(i, j)
            ans += 1

print(ans)