[프로그래머스][Lv. 1] 키패드 누르기 Swift 풀이

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

코딩테스트 연습 - 키패드 누르기

풀이는 추후에 차차 올리도록 하겠습니다...

 

 

 

import Foundation
 
func solution(_ numbers:[Int], _ hand:String) -> String {
    let isLeft: Bool = (hand == "left" ? true : false)
    var leftTh: Int = -1, rightTh: Int = -2
    let coord: [Int:String] = [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", -1:"3 0", -2:"3 2"]
    var answer: String = "", thumb: String = ""
    for num in numbers {
        switch num {
            case 1, 4, 7:
            leftTh = num
            thumb = "L"
            case 3, 6, 9:
            rightTh = num
            thumb = "R"
            case 2, 5, 8, 0:
            if leftTh == -1 && rightTh == -2 {
                if isLeft {
                    leftTh = num
                    thumb = "L"
                } else {
                    rightTh = num
                    thumb = "R"
                }
            } else {
                let numPos: [Int] = coord[num]!.components(separatedBy: " ").map{Int($0)!}
                let leftPos: [Int] = coord[leftTh]!.components(separatedBy: " ").map{Int($0)!}
                let rightPos: [Int] = coord[rightTh]!.components(separatedBy: " ").map{Int($0)!}
                let leftDist: Int = abs(numPos[0] - leftPos[0]) + abs(numPos[1] - leftPos[1])                                                      
                let rightDist: Int = abs(numPos[0] - rightPos[0]) + abs(numPos[1] - rightPos[1])
                if leftDist > rightDist {
                    rightTh = num
                    thumb = "R"
                } else if leftDist < rightDist {
                    leftTh = num
                    thumb = "L"
                } else { // leftDist == rightDist
                    if isLeft {
                        leftTh = num
                        thumb = "L"
                    } else {
                        rightTh = num
                        thumb = "R"                        
                    }
                }
                
            }
            default:
            return "ERROR"
        }
        answer += thumb 
    }
    return answer
}