[Algorithm] 거리두기 확인하기

[hwangJi-dev]

💬 문제

https://school.programmers.co.kr/learn/courses/30/lessons/81302?language=swift

💬 Idea

• 맨해튼 거리가 2 이하인 조건에 부합하는 경우라면 0을, 그렇지 않다면 1을 반환해주자.

<맨해튼 거리가 2 이하인 조건>

인덱스 (y, x)에 P가 있다고 할 때

• 같은 row(행)에 거리가 2 이하인 곳에 P가 있을 때
  • 거리가 1인곳에 P가 있다면 파티션 체크할 필요 없이 false
  • 거리가 2인 곳이라면
    • 가운데에 파티션이 있는지 체크하고 없다면 false
• 같은 col(열)에 거리가 +- 2 이하인 곳에 P가 있을 때
  • 위와 동일
• 좌상 좌하 우상 우하 대각선에 P가 있을 때
  • P X
    X P
  • X P
    P X
  • 해당 2개의 형태가 아니라면 false

💬 풀이

import Foundation

func solution(_ places:[[String]]) -> [Int] {
    var ans: [Int] = []
    
    for i in places {
        var placeArr: [[String]] = []
        i.forEach({ placeArr.append(Array($0).map({ String($0) })) })
        var isRightDistance = true

        for (iIdx, i) in placeArr.enumerated() {
            for (jIdx, j) in i.enumerated() {
                if j == "P" {
                    if !findIfDiaKeepDistance((iIdx, jIdx), placeArr) || !findIfColKeepDistance((iIdx, jIdx), placeArr) || !findIfRowKeepDistance((iIdx, jIdx), placeArr) {
                        isRightDistance = false
                        break
                    }
                }
            }
        }

        ans.append(isRightDistance ? 1 : 0)
    }
    
    return ans
}

// Row (Int, Int)에서 0: y, 1: x
func findIfRowKeepDistance(_ idx: (Int, Int), _ places: [[String]]) -> Bool {
    // 사이에 X 벽이 있을 수 없음
    if idx.1 + 1 <= 4 {
        if places[idx.0][idx.1 + 1] == "P" { return false }
    }
    
    if idx.1 - 1 >= 0 {
        if places[idx.0][idx.1 - 1] == "P" { return false }
    }
    
    if idx.1 + 2 <= 4 {
        if places[idx.0][idx.1 + 2] == "P" && places[idx.0][idx.1 + 1] != "X" { return false }
    }
    
    if idx.1 - 2 >= 0 {
        if places[idx.0][idx.1 - 2] == "P" && places[idx.0][idx.1 - 1] != "X" { return false }
    }
    
    return true
}

// Column
func findIfColKeepDistance(_ idx: (Int, Int), _ places: [[String]]) -> Bool {
    // 사이에 X 벽이 있을 수 없음
    if idx.0 + 1 <= 4 {
        if places[idx.0 + 1][idx.1] == "P" { return false }
    }
    
    if idx.0 - 1 >= 0 {
        if places[idx.0 - 1][idx.1] == "P" { return false }
    }
    
    if idx.0 + 2 <= 4 {
        if places[idx.0 + 2][idx.1] == "P" && places[idx.0 + 1][idx.1] != "X" { return false }
    }
    
    if idx.0 - 2 >= 0 {
        if places[idx.0 - 2][idx.1] == "P" && places[idx.0 - 1][idx.1] != "X" { return false }
    }
    
    return true
}

// Diagonal
func findIfDiaKeepDistance(_ idx: (Int, Int), _ places: [[String]]) -> Bool {
    if idx.0 + 1 <= 4 && idx.1 + 1 <= 4 {
				// 우하
        if places[idx.0 + 1][idx.1 + 1] == "P" {
            if places[idx.0][idx.1 + 1] != "X" || places[idx.0 + 1][idx.1] != "X" {
                return false
            }
        }
    }
    
    if idx.0 - 1 >= 0 && idx.1 - 1 >= 0 {
				// 좌상
        if places[idx.0 - 1][idx.1 - 1] == "P" {
            if places[idx.0 - 1][idx.1] != "X" || places[idx.0][idx.1 - 1] != "X" {
                return false
            }
        }
    }
    
    if idx.0 + 1 <= 4 && idx.1 - 1 >= 0 {
				// 좌하
        if places[idx.0 + 1][idx.1 - 1] == "P" {
            if places[idx.0][idx.1 - 1] != "X" || places[idx.0 + 1][idx.1] != "X" {
                return false
            }
        }
    }
    
    if idx.0 - 1 >= 0 && idx.1 + 1 <= 4 {
				// 우상
        if places[idx.0 - 1][idx.1 + 1] == "P" {
            if places[idx.0 - 1][idx.1] != "X" || places[idx.0][idx.1 + 1] != "X" {
                return false
            }
        }
    }
    
    return true
}

소요시간 : 1시간 30분