12-kjs254

[kjs254]

🔗 문제 링크

안전지대

✔️ 소요된 시간

1h

✨ 수도 코드

1. 문제해석

0으로 구성되어 있고 폭탄은 1로 표시되는 n * n 정사각형 2차원 배열에서 3*3 범위의 폭탄으로부터 안전한 지대의 개수를 반환합니다.

굉장히 간단해서 다양한 풀이방법이 번뜩이는 문제네요.

제가 떠올린 방법은 다음이 있습니다.

1. 폭탄의 개수와 위치만을 파악해서 n²에서 폭탄의 위험지역의 개수를 빼주는 방법

2. 배열을 하나씩 순회하여 폭탄이 발견된다면 상하 좌우 대각선에 -1을 삽입하여 0의 개수를 세는 방법

3. 배열을 순회하여 나의 상하 좌우 대각선에 폭탄이 있는지 체킹하면서 진행하는 방법

수학적으로 접근하는 1번의 연산속도가 빠르지만 따져야하는 조건이 상당한 터라 2번의 방법을 선택했습니다.

3번은 2번과 유사한 방법이지만 괜히 시간복잡도만 늘리는 방법이라 여기서 쓸 방법은 아닌것 같네요.

2차원 배열을 순회하며 폭탄의 상하좌우대각선을 다져보도록 하겠습니다.

2. 수도코드

1. 폭탄의 상하좌우를 -1로 치환한다.

2. borad에서 0의 개수를 센다.

여기서 조건이 몇 개 붙습니다.

board[i][j] 로 배열에 접근을 하려니 폭탄의 위치에 따라 범위를 벗어나는 문제가 있어서 조건을 붙이기보다 그냥 테두리를 감싸주는게 편해보여서 insert(), append() 동원해서 배열을 감싸줬습니다.

n = len(board)
    for i in range(n):
        board[i].insert(0,-1)
        board[i].append(-1)
    board.insert(0,[-1]*(n+2))
    board.append([-1]*(n+2))

테두리와 위험지역에 -1을 삽입한다면 5*5 배열은 다음과 같은 모습이 되어있어야 합니다.

이제 인덱싱 범위를 벗어날 걱정이 없으니 탐색하여 위험지역을 -1로 치환해주고 0의 개수를 반환합니다.

for i in range(n+2):
        for j in range(n+2):
            
            if board[i][j]==1:
                for dx in (-1,0,1):
                    for dy in (-1,0,1):
                        board[i+dy][j+dx] =-1
    
return sum(board, []).count(0) #1차원 배열로 바꾼 후 카운트

하지만 이렇게 풀면 오답입니다.

폭탄이 인접해있다면 1이 -1로 치환되어 폭탄을 숨겨버립니다.

조건을 달면 바로 해결되겠지만 배열의 값이 정수라는 점을 이용하여 연산만으로 해결할 수 있습니다.

board[i+dy][j+dx] = board[i+dy][j+dx]*2 -1

함수 f(x) = 2x-1 에서 f(1) = 1, f(0) = -1이므로 값이 1인 폭탄이 누락될 일이 없어졌습니다.

혹시 0 값이 추가되어 정답에 영향을 줄까 싶지만 f(0.5) = 0이기 때문에 문제없습니다.

전체 코드

def solution(board):
    n = len(board)
    for i in range(n):
        board[i].insert(0,-1)
        board[i].append(-1)
    board.insert(0,[-1]*(n+2))
    board.append([-1]*(n+2))
    
    for i in range(n+2):
        for j in range(n+2):
            
            if board[i][j]==1:
                for dx in (-1,0,1):
                    for dy in (-1,0,1):
                        board[i+dy][j+dx] = board[i+dy][j+dx]*2 -1
    
    return sum(board, []).count(0)

📚 새롭게 알게된 내용

프로그래머스 코딩테스트 입문문제라 그저 재밌게 풀기만 해서 새롭게 알게 된 것은 없네요

다음엔 유사하지만 비슷한 문제로 가지고 와보겠습니다!

[9kyo-hwang]

패딩시켜서 푸는 방법 새롭네요 :) 전 일반적인 방식인 out_of_bound 체크 및 방문 표시 방법으로 풀었었는데...

[9kyo-hwang]

n = len(board)
board = [[-1, *line, -1] for line in board]  # 각 행마다 맨 앞, 맨 뒤에 -1 붙임
board = [[-1] * (n + 2), *board, [-1] * (n + 2)]  # 0행 앞, 마지막 행 뒤에 [-1, ...] 1차원 리스트 붙임

-1 패딩은 요렇게 2줄로 할 수 있답니다 :)

[rivkms]

패딩시켜서 푸는 방법이 되게 인상깊었던 문제였습니다. 새롭게 방법 하나를 배워갈 수 있었습니다.
수학적 연산을 통하여 폭탄의 위치를 조건없이 찾는 방법도 신기했습니다.
예전에 비슷한 문제를 풀던 중 입력을 받을 때 폭탄의 위치를 미리 기억을 해둔 뒤 해당 위치에 대하여 연산을 수행함으로서 소요시간을 줄일 수 있는 방법이 있었던 것으로 기억하는데 이 문제에서도 적용가능할 것 같다는 생각을 하였습니다.

수고하셨습니다. 😁

[YIM2UL2ET]

패딩시켜서 푸는 것도 인상 깊었고, 저는 정수의 특성을 이용해서 조건문 안 붙이고 *2 -1 붙여서 하는건 진짜 놀랍네요.. 이건 저도 앞으로 자주 써먹어야겠습니다. 수고하셨습니다.