프로그래머스 (키패드 누르기, 2020 카카오 인턴십) C++

 

가로로 인접한 키패드는 1만큼 차이가 나고 세로로 인접한 키패드는 3만큼 차이가 나는 이 경우를 어떻게 해결해야 할지 몰라서 고민을 너무 많이 했다. 사실 구글에서 어떤 사람도 고민해서 해결한 결과를 보고 충격에 빠진.... 3으로 나눈 몫과 3으로 나눈 나머지를 더하면 그 차이가 없어진다는 것이 새롭게 바라볼 수 있는 시야를 넓혔다.

 

#include <string>
#include <vector>
#include <iostream>

using namespace std;

string solution(vector<int> numbers, string hand) {
    string answer = "";
    int left_count = 10;
    int right_count = 12;
    
    for(int i = 0 ; i < numbers.size() ; i++) {
        if(numbers[i] == 1 || numbers[i] == 4 || numbers[i] == 7) {
            // 1,4,7 에 해당하는 번호는 L를 push_back 한다.
            answer.push_back('L');
            
            // left_count에는 가장 최근에 사용된 숫자를 저장한다.
            left_count = numbers[i];
        }
        
        else if(numbers[i] == 3 || numbers[i] == 6 || numbers[i] == 9) {
            // 3,6,9 에 해당하는 번호는 R를 push_back 한다.
            answer.push_back('R');
            
            // right_count에는 가장 최근에 사용한 숫자를 저장한다.
            right_count = numbers[i];
        }
        
        else {
            // 숫자 0을 누른 경우 11로 동일한 논리를 적용하기 위해 11로 바꾼다.
            if(numbers[i] == 0) {
                numbers[i] = 11;
            }
            
            // left_dist1은 지금 누른 번호와 가장 최근에 왼손이 누른 번호의 차의 절댓값
            // right_dist1은 지금 누른 번호와 가장 최근에 오른손이 누른 번호의 차의 절댓값
            int left_dist1 = abs(numbers[i] - left_count);
            int right_dist1 = abs(numbers[i] - right_count);
            
            // 가로로 인접한 경우는 차이가 1이고 세로로 인접하면 차이가 3이므로
            // 3으로 나눈 몫과 3으로 나눈 나머지를 더하면 그 차이가 해결된다.(***)
            int left_dist = (left_dist1 / 3) + (left_dist1 % 3);
            int right_dist = (right_dist1 / 3) + (right_dist1 % 3);
            
            // 만약 왼손과 오른손의 거리가 같다면
            if(left_dist == right_dist) {
                // 오른손인 경우
                if(hand == "right") {
                    // R을 push_back하고
                    answer.push_back('R');
                    
                    // right_count에 가장 최근에 사용한 번호를 최신화한다.
                    right_count = numbers[i];
                  
                // 왼손인 경우
                } else if(hand == "left") {
                    // L을 push_back하고
                    answer.push_back('L');
                    
                    // left_count에 가장 최근에 사용한 번호를 최신화한다.
                    left_count = numbers[i];
                }
            }
            // 왼손이 더 가까운 경우
            else if(left_dist < right_dist) {
                // L을 push_back 하고
                answer.push_back('L');
                
                // left_count에 가장 최근에 사용한 번호를 최신화한다.
                left_count = numbers[i];
            }
            // 오른손이 더 가까운 경우
            else if(left_dist > right_dist) {
                // R을 push_back 하고
                answer.push_back('R');
                
                // right_count에 가장 최근에 사용한 번호를 최신화한다.
                right_count = numbers[i];
            }
        }
    }
    
    cout << answer << endl;
    return answer;
}