[Algorithm] 프로그래머스 [카카오 인턴] 키패드 누르기 Level1 (Swift)

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

문제 풀이

입력으로 오는 numbers 배열에 있는 숫자들을 누르는데, 이 때 사용한 손가락을 문자열로 리턴하는 문제.

 

먼저 keypad라는 배열에 키패드 형태를 저장했다.

let keypad = [["1", "2", "3"], ["4", "5", "6"], ["7", "8", "9"], ["*", "0", "#"]]

 

 

키패드 배열은 아래와 같이 저장되어있다.

 

 

 [1, 4, 7]은 왼손 엄지 손가락만 가능하고, [3, 6, 9]는 오른손 엄지 손가락만 가능하므로 따로 배열에 저장해주었다.

let leftOnly = [1, 4, 7] // 왼손 엄지손가락만 가능
let rightOnly = [3, 6, 9] // 오른손 엄지손가락만 가능

 

 

 

왼손 엄지 손가락 위치와 오른손 엄지 손가락 위치를 변수로 저장한다.

• 초기값은 해당 손가락의 시작 위치.
• 왼손 엄지 손가락은 *에서, 오른손 엄지 손가락은 #에서 시작하므로, leftPosition에는 (3, 0)을, rightPosition에는 (3, 2)를 저장한다.

var leftPosition = (3, 0) // 최초 왼손 엄지 손가락 시작 위치
var rightPosition = (3, 2) // 최초 오른손 엄지 손가락 시작 위치

 

 

numbers 배열의 요소를 for문을 통해 돌면서 3가지 경우로 나눠서 처리한다.

 

1. num이 [1, 4, 7]에 해당하는 경우

• 해당 leftPosition의 위치 저장 후, result에 “L”에 추가

2. num이 [3, 6, 9]에 해당하는 경우

• 해당 rightPosition의 위치 저장 후, result에 “R”에 추가

3. 그 외의 경우

num의 위치를 targetPosition에 저장한 후, 현재 leftPosition과 rightPosition의 위치와의 거리를 계산한다.

 

targetPosition과 왼손 엄지 손가락의 거리 : leftToTarget

targetPosition과 오른손 엄지 손가락의 거리 : rightToTarget

 

3-1. leftToTarget이 더 클 경우

• 오른손 엄지 손가락 이동시키고, 해당 rightPosition의 위치 저장 후, result에 “R”에 추가

3-2. rightToTarget이 더 클 경우

• 왼손 엄지 손가락 이동시키고, 해당 leftPosition의 위치 저장 후, result에 “L”에 추가

3-3. 두 거리가 같을 경우

• 왼손 잡이는 result에 “L”을, 오른손 잡이는 result에 “R”을 추가

for 문을 다 돌고 난 후에 result를 리턴하면 정답이다.

 

소스 코드 (Swift)

//
//  키패드누르기.swift
//  Programmers
//
//  Created by 안상희 on 2022/02/06.
//

import Foundation

func solution(_ numbers:[Int], _ hand:String) -> String {
    var result = ""
    let keypad = [["1", "2", "3"], ["4", "5", "6"], ["7", "8", "9"], ["*", "0", "#"]]
    
    let leftOnly = [1, 4, 7] // 왼손 엄지손가락만 가능
    let rightOnly = [3, 6, 9] // 오른손 엄지손가락만 가능
    
    var leftPosition = (3, 0) // 왼손 엄지 손가락 위치 (최초 시작 위치는 *)
    var rightPosition = (3, 2) // 오른손 엄지 손가락 위치 (최초 시작 위치는 #)
    
    for num in numbers {
        // 1, 4, 7은 왼손 엄지손가락만 가능
        if leftOnly.contains(num) {
            result.append("L")
            
            // 왼손 엄지손가락 이동시켜주기
            MoveLeftHand: for i in 0..<4 {
                for j in 0..<3 {
                    if let targetNum = Int(keypad[i][j]), targetNum == num {
                        leftPosition = (i, j) // 위치 저장
                        break MoveLeftHand
                    }
                }
            }
            continue
        }
        
        // 3, 6, 9는 오른손 엄지손가락만 가능
        if rightOnly.contains(num) {
            result.append("R")
            
            // 오른손 엄지손가락 이동시켜주기
            MoveRightHand: for i in 0..<4 {
                for j in 0..<3 {
                    if let targetNum = Int(keypad[i][j]), targetNum == num {
                        rightPosition = (i, j) // 위치 저장
                        break MoveRightHand
                    }
                }
            }
            continue
        }
        
        // 그 외의 경우
        // target의 인덱스 구하기
        var targetIndex = (0, 0)
        GetTarget: for i in 0..<4 {
            for j in 0..<3 {
                if let targetNum = Int(keypad[i][j]), targetNum == num {
                    targetIndex.0 = i
                    targetIndex.1 = j
                    break GetTarget
                }
            }
        }
        
        // left, right에서 target까지의 거리 구하기
        let leftToTarget = abs(leftPosition.0 - targetIndex.0) + abs(leftPosition.1 - targetIndex.1)
        let rightToTarget = abs(rightPosition.0 - targetIndex.0) + abs(rightPosition.1 - targetIndex.1)
        
        if leftToTarget < rightToTarget {
            // 왼쪽 엄지손가락 이동시키기
            leftPosition = (targetIndex.0, targetIndex.1)
            result.append("L")
        } else if leftToTarget > rightToTarget {
            // 오른쪽 엄지손가락 이동시키기
            rightPosition = (targetIndex.0, targetIndex.1)
            result.append("R")
        } else if leftToTarget == rightToTarget {
            if hand == "left" {
                leftPosition = (targetIndex.0, targetIndex.1)
                result.append("L")
            } else {
                rightPosition = (targetIndex.0, targetIndex.1)
                result.append("R")
            }
        }
        
    }
    return result
}