[프로그래머스] 안전지대(Lv.0) - Python

프로그래머스

 

문제 설명

​다음 그림과 같이 지뢰가 있는 지역과 지뢰에 인접한 위, 아래, 좌, 우 대각선 칸을 모두 위험지역으로 분류합니다.

지뢰는 2차원 배열 board에 1로 표시되어 있고 board에는 지뢰가 매설 된 지역 1과, 지뢰가 없는 지역 0만 존재합니다.

지뢰가 매설된 지역의 지도 board가 매개변수로 주어질 때, 안전한 지역의 칸 수를 return하도록 solution 함수를 완성해주세요.

 

제한사항

• board는 n * n 배열입니다.
• 1 ≤ n ≤ 100
• 지뢰는 1로 표시되어 있습니다.
• board에는 지뢰가 있는 지역 1과 지뢰가 없는 지역 0만 존재합니다.

 

---

 

정답 코드

def solution(board):
    N = len(board)
    dx = [-1, 1, 0, 0, -1, -1, 1, 1] # 8방향 탐색
    dy = [0, 0, -1, 1, -1, 1, -1, 1]
    
    # 지뢰 설치
    boom = []
    for i in range(len(board)):
        for j in range(len(board)):
            if board[i][j] == 1:
                boom.append((i,j)) # 지뢰일때의 인덱스 append
                
    # 지뢰가 설치된 곳 주변에 폭탄 설치
    for x, y in boom:
        for i in range(8):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < N and 0 <= ny < N:
                board[nx][ny] = 1

    # 폭탄이 설치되지 않은 곳만 카운팅
    count = 0
    for x in range(N):
        for y in range(N):
            if board[x][y] == 0:
                count += 1
    return count

---

 

코드 풀이

예시처럼 지뢰가 2, 2에 있다고 가정할 때 8방향의 좌표는 아래와 같다.

 

 

따라서 각 좌표 값을 dx, dy에 넣는다.

그리고 폭탄이 설치된 좌표를 찾아 boom에 추가한다.

N은 이후 위험지역을 표시할 때 사용하기 위해 변수에 저장해둔다.

def solution(board):
    N = len(board)
    dx = [-1, 1, 0, 0, -1, -1, 1, 1] # 8방향 탐색
    dy = [0, 0, -1, 1, -1, 1, -1, 1]
    
    # 지뢰 설치
    boom = []
    for i in range(len(board)):
        for j in range(len(board)):
            if board[i][j] == 1:
                boom.append((i,j)) # 지뢰일때의 인덱스 append

 

 

 

boom에 있는 좌표들을 꺼내 dx와 dy 값을 더하여 좌표를 찾는다.

찾은 좌표가 0과 N 사이라면 값을 1로 바꾼다.

if문을 넣은 이유는 폭탄이 한 쪽에 붙어 설치되어 있으면 위험 지역의 갯수가 줄기 때문이다.

if문을 빼버리면 index 오류가 발생한다.

# 지뢰가 설치된 곳 주변에 폭탄 설치
    for x, y in boom:
        for i in range(8):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < N and 0 <= ny < N:
                board[nx][ny] = 1

 

 

 

 

count라는 변수를 선언하여 안전한 구역(값이 1이 아닌 구역)만 카운트한다.

# 폭탄이 설치되지 않은 곳만 카운팅
    count = 0
    for x in range(N):
        for y in range(N):
            if board[x][y] == 0:
                count += 1
    return count