[알고리즘 연습] 자물쇠와 열쇠 (프로그래머스 lv3, 스위프트)

문제

https://programmers.co.kr/learn/courses/30/lessons/60059

코딩테스트 연습 - 자물쇠와 열쇠

풀이

import Foundation

func solution(_ key:[[Int]], _ lock:[[Int]]) -> Bool {
    var key = key
    let n = lock.count
    let m = key.count

    // key를 돌리는 함수
    func rotate() {
        var newKey = [[Int]]()
        for x in 0..<m {
            var temp = [Int]()
            for y in (0..<m).reversed() {
                temp.append(key[y][x])
            }
            newKey.append(temp)
        }
        key = newKey
    }

    // 현재 key를 기준으로 test를 수행하는 함수
    func test() -> Bool {
        for py in 1-m..<n+1 {
            for px in 1-m..<n+1 {
                var mlock = lock
                for y in 0..<m {
                    for x in 0..<m {
                        if py+y >= 0 && px+x >= 0 && py+y < n && px+x < n { 
                            mlock[py+y][px+x] += key[y][x]
                        }
                    }
                }
                if check(mlock) { return true }
            }
        }
        return false
    }

    // 현재 자물쇠값을 확인하는 함수
    func check(_ lock: [[Int]]) -> Bool {
        for y in 0..<lock.count {
            for x in 0..<lock.count {
                if lock[y][x] != 1 {
                    return false
                }
            }   
        }
        return true
    }

    // 현재 상태에서 한번 테스트
    if test() { return true }

    // 돌려서 테스트 3번
    for _ in 1...3 {
        rotate()
        if test() { return true }
    }

    return false
}

• 문제 자체의 난이도는 높지않지만 침착하게 푸는 것이 중요한 문제였다.
• 2차원 배열의 회전 및 좌표 위치의 비교를 잘 생각해야한다.
• 실제 Lock의 좌표를 0부터 N까지라고 생각하면, Key 행동반경은 -(M-1) ~ N+(M-1) 이다.
• 이를 고려해보면 Key의 기준점(Key의 0,0이 배치되는 위치)의 범위는 -(M-1) ~ N+1 이다.
• 이들을 비교해 Key의 좌표와 Lock의 좌표가 일치하는 구간에서는 값을 더하고 모든 좌표가 1인지 확인해줬다.