[Python] 지뢰 찾기 - 백준

해당 글은 백준 1996번 문제 '지뢰 찾기'를 다룬다.

문제

https://www.acmicpc.net/problem/1996

1996번: 지뢰 찾기

 

설명

해당 문제는 지뢰가 없는 곳에서 둘러싼 (상하좌우 + 대각4가지) 총 8가지 구역 중에 지뢰가 있는 부분을 다 더해줘서 업데이트 시켜주면 된다. 만약 8구역의 총합이 10 이상이면 M으로 출력하고, 지뢰가 있는 부분은 *로 출력하면 되는 간단한 구현 문제이다.

 

 

코드

n = int(input())
arr = []
for i in range(n):
    arr.append(list(input()))

dx = [-1, 1, 0, 0, -1, -1, 1, 1]
dy = [0, 0, 1, -1, -1, 1, -1, 1]

result = [[0 for _ in range(n)] for _ in range(n)]

for i in range(n):
    for j in range(n):
        if arr[i][j] == '.':
            count = 0
            for k in range(8):
                nx = i + dx[k]
                ny = j + dy[k]
                if nx >= 0 and nx < n and ny >= 0 and ny < n and arr[nx][ny] != '.':
                    count += int(arr[nx][ny])
            result[i][j] = str(count) if count < 10 else 'M'
        else:
            result[i][j] = '*'

for i in range(n):
    print(''.join(result[i]))