[프로그래머스] 안전지대

다른 공부를 하면서 간단하게 하루에 한 문제씩 풀려고 했습니다. 그러다가 안전지대 문제를 풀게 되었고 문제를 읽어보니, 2차원 배열에서 폭탄이 설치된 위치 주변에 폭탄을 설치하고, 폭탄이 설치되지 않은 지역의 개수를 반환하는 문제였습니다. 이런 유형의 문제를 여러 번 풀어봐서 익숙해져 있어서 도전하게 되었습니다.처음에 작성한 코드로는 주어진 기본 테스트 케이스는 통과했지만, 제출할 때는 절반 이상의 테스트 케이스가 실패하여 어리둥절했습니다...??? 😮

 

어라라..? 😭

---

1. 문제 링크 

 

프로그래머스

https://school.programmers.co.kr/learn/courses/30/lessons/120866

 

2. 코드

func solution(_ board:[[Int]]) -> Int {
    let row = [-1,-1,-1,0,0,1,1,1]
    let col = [-1,0,1,-1,1,-1,0,1]
    let size = board.count
    var tempArr = board

    func setBombArea(i:Int, j:Int){
        for k in 0..<8{
            if (i + row[k] >= 0 && i + row[k] < size) && (j + col[k] >= 0 && j + col[k] < size){
                tempArr[i+row[k]][j+col[k]] = 1
            }
        }

    }
    // 폭탄 찾고 설치하기 부분
    for i in 0..<size{
        for j in 0..<size{
            if board[i][j] != 1{
                continue
            }
            setBombArea(i: i, j: j)
        }
    }
    let safeArea = tempArr.map({ $0.filter({ $0 == 0 }).count }).reduce(0,+)
    return safeArea
}

 

3.  코드 설명

먼저 row, col 변수는 위, 아래, 양 옆, 대각선 방향을 탐색하기 위해 선언되었습니다. 반복문에서 인덱스 값에 더해 해당 방향의 위치를 구할 수 있습니다.
size 변수는 주어진 배열 board의 크기를 나타내는 값으로 "board는 n * n 크기의 배열"이라고 문제에 명시되어 있어 고정값으로 설정하였습니다.

이중 for-in문을 사용하여 각각의 인덱스에 접근하였습니다. 만약 해당 인덱스의 값이 1이 아니라면 continue를 사용하여 다음 순서로 넘어갑니다. 그렇지 않고 값이 1이라면 setBombArea 메서드에 현재 인덱스 위치 i, j를 전달하여 위, 아래, 양 옆, 대각선 방향을 탐색하도록 만들었습니다. 탐색할 때는 각 방향으로 더해진 값들이 board의 배열 범위 내에 속하는지 확인합니다.

만약 해당 위치가 배열 범위 내에 속한다면 tempArr에 1을 설정하여 폭탄을 설치합니다. 이후 tempArr에서 값이 0인 원소들의 개수를 세어 safeArea에 저장합니다.

 

 

---

문제를 맞히기 위해 코드를 수정한 과정에서 두 가지 오류를 발견했습니다. 첫째로 배열 범위를 잘못 고려하여 0을 포함시키지 않았던 것입니다. 둘째로 메서드에 전달되는 매개변수의 변수명이 중복되는 문제가 있었습니다. 틀렸을 때 당시에 설마 내가 문제를 잘못 이해했을까?? 다시 읽어보고 내가 발견하지 못하는 다른 예외 처리가 있을까 생각했었습니다 ㅋㅋ;;😅