[프로그래머스/Lv.1] 키패드 누르기(2020 KAKAO INTERNSHIP) - Java

📌 문제

프로그래머스

📌 풀이

키패드 간 거리를 구하는 방법과 *, 0, #을 처리하는 방법을 몰라 답을 보고 푼 문제이다. 먼저 *, 0, #은 다음과 같이 차례대로 10, 11, 12로 설정한 후 문제를 푼다.

1	2	3
4	5	6
7	8	9
*(10)	0(11)	#(12)

• numbers 배열을 돌면서 순서대로 번호를 누르며 왼손으로 눌렀는지 오른손을 눌렀는지 판단한다.
• 누른 번호가 1, 4, 7 이라면
  • 왼손으로 키패드를 누르고 왼손을 누른 자리로 이동시킨다.
• 누른 번호가 3, 6, 9 라면
  • 오른손으로 키패드를 누르고 오른손을 누른 자리로 이동시킨다.
• 그 외 번호라면
  • 왼손으로부터의 거리와 오른손으로부터의 거리를 구한다.
  • 왼손으로부터의 거리가 더 크다면
    • 오른손으로 키패드를 누르고 오른손을 누른 자리로 이동시킨다.
  • 오른손으로부터의 거리가 더 크다면
    • 왼손으로 키패드를 누르고 왼손을 누른 자리로 이동시킨다.
  • 거리가 같다면
    • hand의 값에 따라 처리한다.

흐름만 보면 어렵지 않지만 "왼손으로부터의 거리와 오른손으로부터의 거리를 구한다." 라는 부분을 해결하지 못해서 어려웠다. 이 부분은 다음과 같이 구하면 된다.

int leftDist = Math.abs(num - left) / 3 + Math.abs(num - left) % 3;
int rightDist = Math.abs(num - right) / 3 + Math.abs(num - right) % 3;

📌 다른 풀이 -  좌표로 구하기

다른 코드들을 살펴보니 키패드를 좌표로 놓고 구하는 풀이도 있어 기록해놓으려고 한다. 이 풀이는 메서드를 나눠 작성했기 때문에 더 가독성있고 깔끔해보인다! 이 풀이에서는 다음과 같은 새로운 클래스를 만들어 해결했다.

• Position 클래스
  • 손의 좌표를 저장할 Position 생성자 메서
  • 키패드를 누르는 손을 구하는 getFinger() 메서드
  • 키패드와 손 사이의 거리를 구하는 getDistance() 메서드

먼저 간단한 Position 클래스와 생성자 함수를 만들었다.

class Position {
    int row; 
    int col;
    
    Position(int row, int col) {
        this.row = row;
        this.col = col;
    }
}

 

solution 함수에 만들어둔 Position 클래스를 이용해 왼손의 위치를 저장할 left, 오른손의 위치를 저장할 right, 누를 번호의 위치를 저장할 numPos 변수를 만든다. 그리고 나서 left와 right를 초기화한다. 왼손은 *, 즉 좌표상 (3, 0)에 위치하고 오른손은 #, 즉 좌표상 (3, 2)에 위치한다.

Position left;
Position right;
Position numPos;
    
public String solution(int[] numbers, String hand) {
    StringBuilder answer = new StringBuilder();

    left = new Position(3, 0);
    right = new Position(3, 2);
    
    return answer.toString();
}

 

헷갈리면 아래와 같이 좌표를 적은 그림을 그려놓고 생각해보자.

(0, 0)	(0, 1)	(0, 2)
(1, 0)	(1, 1)	(1, 2)
(2, 0)	(2, 1)	(2, 2)
(3, 0)	(3, 1)	(3, 2)

 

그리고나서 numbers 배열을 돌면서 숫자를 누를 손가락을 정하고, answer에 정한 손가락을 저장한 후 손가락 위치를 이동하는 과정이다. numPos의 좌표를 다음과 같은 식으로 구한다.(위에서 거리를 구했던 방식과 유사하다.)

for (int num : numbers) {
    numPos = new Position((num - 1) / 3, (num - 1) % 3); // numPos의 좌표 구하기
    if (num == 0) numPos = new Position(3, 1); // 번호가 0이라면 좌표를 (3, 1)로 넣어주기
    String finger = numPos.getFinger(hand); // 번호를 누를 손가락 정하기
            
    answer.append(finger);
            
    if (finger.equals("L")) left = numPos;
    else right = numPos;
}

 

다음은 Position 클래스 안에 있는 getFinger() 메서드와  getDistance() 메서드이다. hand를 이용해 finger를 초기화하고 있기 때문에 if문을 그냥 빠져나오더라도 상관없다. col이 0이면 1, 4, 7이기 때문에 왼손으로 누르고, col이 2라면 3, 6, 9이기 때문에 오른손으로 누른다. 둘 다 아니라면 else문으로 나와 각 거리를 구해 더 가까운 쪽의 손을 반환한다. 참고란에 있는 유튜브 영상을 보면 더 이해가 잘 갈 것이다.

public String getFinger(String hand) {
    String finger = hand.equals("right") ? "R" : "L";
        
    if (this.col == 0) finger = "L";
    else if (this.col == 2) finger = "R";
    else {
        int leftDist = left.getDistance(this);
        int rightDist = right.getDistance(this);
            
        if (leftDist < rightDist) finger = "L";
        else if (rightDist < leftDist) finger = "R";
    }
        
    return finger;
}
    
public int getDistance(Position p) {
    return Math.abs(this.row - p.row) + Math.abs(this.col - p.col);
}

 

📌 전체 코드

첫 번째 코드

class Solution {
    public String solution(int[] numbers, String hand) {
        StringBuilder answer = new StringBuilder();
        int left = 10;
        int right = 12;
        for (int num : numbers) {
            if (num == 1 || num == 4 || num == 7) {
                answer.append('L');
                left = num;
            } else if (num == 3 || num == 6 || num == 9) {
                answer.append('R');
                right = num;
            } else {
                if (num == 0) num = 11;
                int leftDist = Math.abs(num - left) / 3 + Math.abs(num - left) % 3;
                int rightDist = Math.abs(num - right) / 3 + Math.abs(num - right) % 3;
                
                if (leftDist > rightDist) {
                    answer.append('R');
                    right = num;
                } else if (leftDist < rightDist) {
                    answer.append('L');
                    left = num;
                } else {
                    if (hand.equals("right")) {
                        answer.append('R');
                        right = num;
                    } else {
                        answer.append('L');
                        left = num;
                    }
                }
            }
        }
        
        return answer.toString();
    }
}

두 번째 코드

class Solution {
    Position left;
    Position right;
    Position numPos;
    
    public String solution(int[] numbers, String hand) {
        StringBuilder answer = new StringBuilder();
        // 1. 왼손, 오른손 위치 초기화
        left = new Position(3, 0);
        right = new Position(3, 2);
        
        for (int num : numbers) {
            // 2. 숫자를 누를 손가락 정하기
            numPos = new Position((num - 1) / 3, (num - 1) % 3);
            if (num == 0) numPos = new Position(3, 1);
            String finger = numPos.getFinger(hand);
            
            //3. 정해진 손가락을 answer에 달고, 손가락 위치 이동
            answer.append(finger);
            
            if (finger.equals("L")) 
                left= numPos;
            else 
                right = numPos;
        }
        
        return answer.toString();
    }

class Position {
    int row; 
    int col;
    
    Position(int row, int col) {
        this.row = row;
        this.col = col;
    }
    
    public String getFinger(String hand) {
        String finger = hand.equals("right") ? "R" : "L";
        
        if (this.col == 0) finger = "L";
        else if (this.col == 2) finger = "R";
        else {
            int leftDist = left.getDistance(this);
            int rightDist = right.getDistance(this);
            
            if (leftDist < rightDist)
                finger = "L";
            else if (rightDist < leftDist)
                finger = "R";
        }
        
        return finger;
    }
    
    public int getDistance(Position p) {
        return Math.abs(this.row - p.row) + Math.abs(this.col - p.col);
    }
}
    
}

 

참고

https://youtu.be/jFv2LRVCK6s?si=gn32MTRzCKMdjXxN