[PROGRAMMERS] 키패드 누르기 - JAVA

1. 문제

프로그래머스

이 전화 키패드에서 왼손과 오른손의 엄지손가락만을 이용해서 숫자만을 입력하려고 합니다.
맨 처음 왼손 엄지손가락은 * 키패드에 오른손 엄지손가락은 # 키패드 위치에서 시작하며, 엄지손가락을 사용하는 규칙은 다음과 같습니다.

1. 엄지손가락은 상하좌우 4가지 방향으로만 이동할 수 있으며 키패드 이동 한 칸은 거리로 1에 해당합니다.
2. 왼쪽 열의 3개의 숫자 1, 4, 7을 입력할 때는 왼손 엄지손가락을 사용합니다.
3. 오른쪽 열의 3개의 숫자 3, 6, 9를 입력할 때는 오른손 엄지손가락을 사용합니다.
4. 가운데 열의 4개의 숫자 2, 5, 8, 0을 입력할 때는 두 엄지손가락의 현재 키패드의 위치에서 더 가까운 엄지손가락을 사용합니다.
4-1. 만약 두 엄지손가락의 거리가 같다면, 오른손잡이는 오른손 엄지손가락, 왼손잡이는 왼손 엄지손가락을 사용합니다.

순서대로 누를 번호가 담긴 배열 numbers, 왼손잡이인지 오른손잡이인 지를 나타내는 문자열 hand가 매개변수로 주어질 때, 각 번호를 누른 엄지손가락이 왼손인 지 오른손인 지를 나타내는 연속된 문자열 형태로 return 하도록 solution 함수를 완성해주세요.

---

2. 풀이

Solution

left, right에는 현재 오른쪽과 왼쪽 엄지손가락 위치를 저장한다. 초기값은 -1이다. 

isLeft에는 왼손잡이인지, 오른손잡이인지를 저장한다. 왼손잡이라면 true다. 

lines라는 HashMap은 키패드 숫자와 그 숫자가 몇 번째 줄에 있는지를 저장한다.

0번째 줄	1	2	3
1번째 줄	4	5	6
2번째 줄	7	8	9
3번째 줄	*	0	#

 

StringBuilder에 numbers 요소를 하나씩 탐색하면서 왼쪽인지 오른쪽 손인지 체크한 후 그 값을 더한다.

sb를 String으로 변환해서 반환한다.

    int left = -1, right = -1;
    boolean isLeft;
    Map<Integer,Integer> lines = new HashMap<>();
    
    public String solution(int[] numbers, String hand) {
        lines.put(0,3);
        lines.put(-1,3);
        for(int i=1;i<10;i+=3){
            lines.put(i, i/3);
            lines.put(i+1,i/3);
            lines.put(i+2,i/3);
        }

        isLeft = hand.equals("left");
        
        StringBuilder sb = new StringBuilder();
        for(int num:numbers){
            sb.append(select(num));
        }
        
        return sb.toString();
    }

 

손 선택

number를 3으로 나눴을때 나머지가 1이라면 1,4,7에 해당된다. 이 수는 왼손으로 눌러야 하기 때문에 left에 현재 숫자를 저장하고 'L'을 리턴한다.

number를 3으로 나눴을때 나머지가 0이면 number==0인 경우를 제외하고 3,6.9다. 이 수는 오른손 수이므로 right에 number를 저장하고 'R'을 리턴한다.

 

ldist에 왼손으로 number을 눌렀을 때의 거리, rdist에 오른손으로 number을 눌렀을 때의 거리를 저장한다.

ldist가 더 작거나 ldist==rdist이지만 왼손잡이인 경우 left에 number을 저장하고 'L'을 리턴한다.

그 외의 경우 right에 number을 저장하고 'R'을 리턴한다.

    public char select(int number){
        if(number%3==1) {
            left = number;
            return 'L';
        }else if(number != 0 && number%3==0){
            right = number;
            return 'R';
        }
        
        int ldist = calc(number,true);
        int rdist = calc(number,false);
        
        if(ldist<rdist || (ldist == rdist && isLeft)){
            left = number;
            return 'L';
        }else{
            right = number;
            return 'R';
        }
    }

 

거리 계산

왼손	왼손,오른손	오른손
1	2	3
4	5	6
7	8	9
*	0	#

이동하려고 하는 수가 2,5,8,0에 해당될 때만 이 메서드가 실행된다.

왼손은 왼쪽 두 줄, 오른손은 오른쪽 두 줄로만 이동 가능하기 때문에 거리 계산이 복잡하지 않다. 이동하려는 숫자와 현재 숫자의 세로 줄 차이 + 가로로 최대 한 번 이동하면 끝이다.

 

현재 손가락 위치가 위에서부터 몇 번째 줄인지 파악하고, (2,5,8,0)과 같은 세로 줄에 있다면 0, 같은 줄에 없다면 1을 먼저 answer에 저장한다. 가로 이동 횟수를 미리 더하는 것이다.

 

answer에 (현재 손가락의 세로줄 위치 - 누르려는 숫자의 세로줄 위치) 절댓값을 더한다.

answer을 반환한다. 

    public int calc(int number, boolean lefthand){
        int answer = 1;
        int pos = lefthand? left:right;
        
        if(pos == 0 || pos%3==2) answer-=1; 
        
        answer += Math.abs(lines.get(pos)-lines.get(number));        
        return answer;
    }

 

 전체 코드

import java.util.*;

class Solution {
    int left = -1, right = -1;
    boolean isLeft;
    Map<Integer,Integer> lines = new HashMap<>();
    
    public String solution(int[] numbers, String hand) {
        lines.put(0,3);
        lines.put(-1,3);
        for(int i=1;i<10;i+=3){
            lines.put(i, i/3);
            lines.put(i+1,i/3);
            lines.put(i+2,i/3);
        }

        isLeft = hand.equals("left");
        
        StringBuilder sb = new StringBuilder();
        for(int num:numbers){
            sb.append(select(num));
        }
        
        return sb.toString();
    }
    
    public char select(int number){
        if(number%3==1) {
            left = number;
            return 'L';
        }else if(number != 0 && number%3==0){
            right = number;
            return 'R';
        }
        
        int ldist = calc(number,true);
        int rdist = calc(number,false);
        
        if(ldist<rdist || (ldist == rdist && isLeft)){
            left = number;
            return 'L';
        }else{
            right = number;
            return 'R';
        }
    }
    
    public int calc(int number, boolean lefthand){
        int answer = 1;
        int pos = lefthand? left:right;
        
        if(pos == 0 || pos%3==2) answer-=1; 
        
        answer += Math.abs(lines.get(pos)-lines.get(number));        
        return answer;
    }
}

---

3. 결과