[Python/파이썬] 백준 4396번 지뢰 찾기

4396번: 지뢰 찾기

 

문제

지뢰찾기는 n × n 격자 위에서 이루어진다. m개의 지뢰가 각각 서로 다른 격자 위에 숨겨져 있다. 플레이어는 격자판의 어느 지점을 건드리기를 계속한다. 지뢰가 있는 지점을 건드리면 플레이어가 진다. 지뢰가 없는 지점을 건드리면, 그곳의 상하좌우 혹은 대각선으로 인접한 8개의 칸에 지뢰가 몇 개 있는지 알려주는 0과 8 사이의 숫자가 나타난다. 완전히 플레이되지 않은 게임에서 일련의 동작이 아래에 나타나 있다.

  

여기서, n은 8이고, m은 10이며, 빈 칸은 숫자 0을 의미하고, 올라가 있는 칸은 아직 플레이되지 않은 위치이며, 별표 모양(*)과 닮은 그림은 지뢰를 의미한다. 맨 왼쪽의 그림은 일부만이 플레이된 게임을 나타낸다. 첫 번째 그림에서 두 번째 그림으로 오면서, 플레이어는 두 번의 이동을 시행해서, 두 번 다 안전한 곳을 골랐다. 세 번째 그림을 볼 때 플레이어는 운이 썩 좋지는 않았다. 지뢰가 있는 곳을 골라서 게임에서 졌다. 플레이어는 m개의 열리지 않은 칸을 남길 때까지 계속해서 안전한 곳을 고르면 이긴다. 그 m개의 칸은 반드시 지뢰이다.

당신이 할 일은 일부가 플레이된 게임의 정보를 읽어 해당하는 격자를 출력하는 것이다.

입력

첫 번째 줄에는 10보다 작거나 같은 양의 정수 n이 입력된다. 다음 n개의 줄은 지뢰의 위치를 나타낸다. 각각의 줄은 n개의 문자를 사용하여 한 행을 나타낸다. 온점(.)은 지뢰가 없는 지점이며 별표(*)는 지뢰가 있는 지점이다. 다음 n개의 줄에는 길이가 n인 문자열이 입력된다. 이미 열린 칸은 영소문자 x로, 열리지 않은 칸은 온점(.)으로 표시된다. 예제 입력은 문제 설명에서의 가운데 그림과 상응한다.

출력

출력은 각각의 위치가 정확하게 채워진 판을 표현해야 한다. 지뢰가 없으면서 열린 칸에는 0과 8 사이의 숫자가 있어야 한다. 지뢰가 있는 칸이 열렸다면 지뢰가 있는 모든 칸이 별표(*)로 표시되어야 한다. 다른 모든 지점은 온점(.)이어야 한다.

코드

n = int(input())

# 지뢰 위치 입력
bomb = []
for i in range(n):
    arr = input()
    for j in range(n):
        if arr[j] == '*':   # 지뢰 위치 저장
            bomb.append([i, j])
    
# 주변에 몇 개의 지뢰가 있는지
def get_bomb_num(x, y):
    dx = [-1, -1, -1, 0, 1, 1, 1, 0]
    dy = [-1, 0, 1, 1, 1, 0, -1, -1]
    
    res = 0
    for i in range(8):
        nx = x + dx[i]
        ny = y + dy[i]
        if 0 <= nx < n and 0 <= ny < n:
            if [nx, ny] in bomb:
                res += 1
    return res

# 열린 칸 입력        
board = [[-1] * n for _ in range(n)]
bomb_open = False
for i in range(n):
    row = input()
    for j in range(n):
        # 열린 칸
        if row[j] == 'x':   
            if [i, j] not in bomb:  # 지뢰가 아닌 칸
                board[i][j] = get_bomb_num(i, j)
            else:                   # 지뢰인 칸 -> 지뢰 있는 모든 칸 열어야 함
                if not bomb_open:
                    for bx, by in bomb:
                        board[bx][by] = '*'
                    bomb_open = True
        # 안 열린 칸
        else:               
            if [i, j] not in bomb:  # 지뢰가 아닌칸
                board[i][j] = '.'
            else:                   # 지뢰인 칸
                if bomb_open:
                    continue        # 이미 지뢰 처리 완료
                else:
                    board[i][j] = '.'
                    
for i in range(n):
    for j in range(n):
        print(board[i][j], end='')
    print()