[swift] 프로그래머스 - 키패드 누르기

프로그래머스

---

문제 푸는 데 있어 도움이 되도록 나의 풀이와 개선된 풀이를 올립니다.
또한 풀이 후 다른 사람의 풀이를 보고 참고할만한 풀이도 올립니다.
- 문제에 따라 나의 풀이만 있을 수 있습니다.
- 해당 풀이들은 풀이 중 하나일 뿐 최선의 풀이는 아닐 수 있습니다.

 

나의 풀이

import Foundation

func solution(_ numbers:[Int], _ hand:String) -> String {
    var result = [String]()
    var state = [0,0]
    var lastLeft = 1
    var lastRight = 1
    var center = ["0":0, "8":1, "5":2, "2":3]
    
    func calculate(num: String) {
        var left = abs(center[num]! - state[0]) + lastLeft
        var right = abs(center[num]! - state[1]) + lastRight
        if num == "0" {
            if left < right { result.append("L"); state[0] = 0; lastLeft = 0; return }
            if left > right { result.append("R"); state[1] = 0; lastRight = 0; return }
            if hand == "left" { result.append("L"); state[0] = 0; lastLeft = 0; return }
            else { result.append("R"); state[1] = 0; lastRight = 0; return }
        }
        if num == "8" {
            if left < right { result.append("L"); state[0] = 1; lastLeft = 0; return }
            if left > right { result.append("R"); state[1] = 1; lastRight = 0; return }
            if hand == "left" { result.append("L"); state[0] = 1; lastLeft = 0; return }
            else { result.append("R"); state[1] = 1; lastRight = 0; return }
        }
        if num == "5" {
            if left < right { result.append("L"); state[0] = 2; lastLeft = 0; return }
            if left > right { result.append("R"); state[1] = 2; lastRight = 0; return }
            if hand == "left" { result.append("L"); state[0] = 2; lastLeft = 0; return }
            else { result.append("R"); state[1] = 2; lastRight = 0; return }
        }
        if num == "2" {
            if left < right { result.append("L"); state[0] = 3; lastLeft = 0; return }
            if left > right { result.append("R"); state[1] = 3; lastRight = 0; return }
            if hand == "left" { result.append("L"); state[0] = 3; lastLeft = 0; return }
            else { result.append("R"); state[1] = 3; lastRight = 0; return }
        }
    }
    
    numbers.forEach { i in
        var num = String(i)
        switch num {
        case "7": result.append("L"); state[0] = 1; lastLeft = 1
        case "4": result.append("L"); state[0] = 2; lastLeft = 1
        case "1": result.append("L"); state[0] = 3; lastLeft = 1
        case "9": result.append("R"); state[1] = 1; lastRight = 1
        case "6": result.append("R"); state[1] = 2; lastRight = 1
        case "3": result.append("R"); state[1] = 3; lastRight = 1
        case "0": calculate(num: num)
        case "8": calculate(num: num)
        case "5": calculate(num: num)
        case "2": calculate(num: num)
        default :
            break
        }
    }
    return result.joined()
}

 

* 해당 풀이는 비추천입니다. 참고할만한 풀이를 찾으신다면 밑에 코드를 참고하세요.

 

숫자 1,4,7은 무조건 "L"을 return

숫자 3,6,9는 무조건 "R"을 return

하게 된다.

 

중요한건

숫자 0,2,5,8을 처리하는건데

두 손가락 중 마지막 위치가 숫자에 가까우면

가까운 손가락이 숫자를 누르게 해줘야한다.

 

몇 가지 변수를 지정해줬는데

 

state[0] 에는 왼쪽 손가락의 층을 (7이면 1층, 4면 2층, 1이면 3층)

마찬가지로

state[1] 에는 오른쪽 손가락의 층을 저장하게 된다.

 

lastLeft와 lastRight는

1일 때는 왼쪽이나, 오른쪽에 마지막 손가락이 있었던거고

0일 때는 손가락이 가운데 있었던 정보를 저장해놨다.

 

만약 가운데 번호를 누를 때

마지막 손가락 위치가 가운데였다면

1만큼의 움직임을 최소화하기 때문에

그 값을 적용하기 위해서다.

 

깔끔한 코드가 아니어서 직관적이지 않지만

나름 상하좌우 위치를 구할 수 있게 된다.

 

숫자마자 해당 위치 조건을 따져

답을 구할 수 있게 된다.

 

다시 말하지만

좋은코드는 아니라

참고하시는건 비추입니다.

 

 

 

---

참고할만한 풀이

import Foundation

func solution(_ numbers:[Int], _ hand:String) -> String {
    var result:String = ""
    let dic:[Int:[Int]] = [
        1:[0,0],
        2:[0,1],
        3:[0,2],
        4:[1,0],
        5:[1,1],
        6:[1,2],
        7:[2,0],
        8:[2,1],
        9:[2,2],
        0:[3,1],
    ]
    var lastL:[Int] = [3,0]
    var lastR:[Int] = [3,2]

    for number in numbers {
        if number == 1 || number == 4 || number == 7 {
            lastL = dic[number]!
            result += "L"
        } else if number == 3 || number == 6 || number == 9 {
            lastR = dic[number]!
            result += "R"
        } else {
            var sizeL = abs(lastL[0]-dic[number]![0])+abs(lastL[1]-dic[number]![1])
            var sizeR = abs(lastR[0]-dic[number]![0])+abs(lastR[1]-dic[number]![1])
            if sizeL < sizeR || (sizeL == sizeR && hand == "left") {
                lastL = dic[number]!
                result += "L"
            } else {
                lastR = dic[number]!
                result += "R"
            }
        }
    }
    return result
}

 

알고리즘은 같다고 할 수 있지만

훨씬 깔끔하게 짠 코드입니다.

 

숫자의 위치정보를 딕셔너리로, 값에는 좌표배열을 넣었습니다.

 

0,2,5,8을 처리하는 부분만

한번 생각해보면서 보시면 될거같습니다.

 

---