[BOJ/Python] 백준 16918번 - 봄버맨

백준 #16918 봄버맨

16918번: 봄버맨

import sys
input = sys.stdin.readline

r, c, n = map(int, input().split())   # r: 행, c: 열, n: 초
initial_grid = [list(input().strip()) for _ in range(r)]

def bomb(grid):
    current_grid = [['O' for _ in range(c)] for _ in range(r)]
    for i in range(r):
        for j in range(c):
            if grid[i][j] == 'O':
                current_grid[i][j] = '.'
                for dx, dy in (0, 1), (0, -1), (1, 0), (-1, 0):
                    x, y = i + dx, j + dy
                    if 0 <= x < r and 0 <= y < c:
                        current_grid[x][y] = '.'
    return current_grid

if n == 1:   # 초기 상태
    for i in range(r):
        print(''.join(initial_grid[i]))
elif n % 2 == 0:   # 짝수 초에는 전체에 폭탄
    for i in range(r):
        print('O' * c)
elif n % 4 == 3:   # 초기 상태에서 폭탄 터진 상태
    result_grid = bomb(initial_grid)
    for i in range(r):
        print(''.join(result_grid[i]))
elif n % 4 == 1:
    result_grid = bomb(initial_grid)
    result_grid = bomb(result_grid)
    for i in range(r):
        print(''.join(result_grid[i]))

 

격자판의 모양은 규칙적인 형태로 나타난다! 4로 나눴을 때 나머지에 따라 바뀌는데,

• 나머지가 1일 때 (1초 제외) : 초기 상태에서 폭탄이 두 번 터진 상태
• 나머지가 짝수일 때 : 전체에 폭탄이 깔린 상태
• 나머지가 3일 때 : 초기 상태에서 폭탄이 한 번 터진 상태

이렇게 나뉜다. 1초일 때는 초기 상태를 출력하면 된다. 그리고 bomb 함수는 폭탄이 터진 후 상태를 구하는 함수로, 초기 상태에서 0인 곳이 있다면 0으로 세팅 돼 있는 새 그리드에 해당 위치와 상하좌우를 . 으로 세팅한다.

예제로 준 문제를 시간초에 따른 상태를 그려보았을 때 1초와 5초, 9초 모두 같은 상태로 나와서 나머지가 1일 때 초기 상태로 출력하는 코드로 구현을 했었는데 다르게 나오는 예제가 있다고 한다. 예제는 찾지 못 함 ,, 1초와 다르게 5초, 9초는 초기 상태에서 폭탄이 두 번 터진 상태로 다르게 구현해야 하는 게 포인트 ..!!

힘들었던 문제 :(