BOJ 15240 - Paint bucket

문제

백준 온라인 저지 - 15240번

풀이 과정

pixels 라는 이차원 배열을 생성한 다음, 시작 지점에서 BFS 를 통해 방문을 수행합니다.
이때 방문처리 하기 전에 이전 위치의 픽셀 값을 저장해서 비교하는 것에 유의합니다.

코드

import sys
from collections import deque


dr = [0, 0, 1, -1]
dc = [1, -1, 0, 0]


def bfs(r, c, color, pixels):
    q = deque()
    visit = [[0] * C for _ in range(R)]

    q.append([r, c])
    visit[r][c] = color

    while q:
        r, c = q.popleft()
        before = pixels[r][c]
        pixels[r][c] = color

        for i in range(4):
            nr, nc = r + dr[i], c + dc[i]
            if 0 <= nr < R and 0 <= nc < C:
                if pixels[nr][nc] == before and not visit[nr][nc]:
                    visit[nr][nc] = 1
                    q.append([nr, nc])

    return pixels


def solution():
    answer = bfs(Y, X, K, pixels)

    for r in range(R):
        print(''.join(map(str, answer[r])))


if __name__ == '__main__':
    R, C = list(map(int, sys.stdin.readline().split()))
    pixels = [list(sys.stdin.readline().strip()) for _ in range(R)]
    Y, X, K = list(map(int, sys.stdin.readline().split()))

    solution()