[프로그래머스] 키패드 누르기

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

프로그래머스

---

python

def solution(numbers, hand):
    answer = ''
    
    # 편의상 * -> 10, 0 -> 11, # -> 12 로 바꾸어 사용
    
    # 좌표상 위치 저장
    location = dict()
    for i in range(1,13):
        location[i] = [(i-1)//3, (i-1)%3]
    
    # 초기 위치
    left = 10
    right = 12
    
    for i in numbers:
        if i == 0: # 편의상 0 -> 11 로 바꾸어 사용
            i = 11
        
        # 왼쪽줄은 왼손으로 누른다
        if i in [1,4,7]:
            answer += "L"
            left = i
        # 오른쪽줄은 오른손으로 누른다
        elif i in [3,6,9]:
            answer += "R"
            right = i
        else:
            # 거리 계산
            ldist = abs(location[left][0] - location[i][0]) + abs(location[left][1] - location[i][1])
            rdist = abs(location[right][0] - location[i][0]) + abs(location[right][1] - location[i][1]) 
            
            # 키패드 누를 손가락 결정
            if ldist > rdist:
                answer += "R"
                right = i
            elif ldist == rdist:
                if hand == "right":
                    answer += "R"
                    right = i
                else:
                    answer += "L"
                    left = i
            else:
                answer += "L"
                left = i

    return answer

 

 

c++

#include <string>
#include <vector>
#include <map>

using namespace std;

string solution(vector<int> numbers, string hand) {

	// 좌표 위치 저장
	// 편의상 * -> 10, 0 -> 11, # -> 12 로 바꿈
    map<int,pair<int,int>> location;
    for(int i=1; i<13; i++){
        location.insert({i, make_pair((i-1)/3, (i-1)%3)});
    }
    
    string answer = "";
    
    // 왼손, 오른손 위치 초기화
    int left = 10;
    int right = 12;
    
    for(int i=0; i<numbers.size(); i++){
        int cur = numbers[i];
        
        if(cur==0) cur=11; // 편의상 0 -> 11로 바꿈
        
        if(location[cur].second == 0){ // 왼쪽줄이면 왼손
            answer += "L";
            left = cur;
        }
        else if (location[cur].second == 2){ // 오른쪽줄이면 오른손
            answer += "R";
            right = cur;
        }
        else{
            // 거리 계산
            int ldist = abs(location[left].first - location[cur].first) + abs(location[left].second - location[cur].second);
            int rdist = abs(location[right].first - location[cur].first) + abs(location[right].second - location[cur].second);
            
            // 손가락 결정하기
            if (ldist > rdist){
                answer += "R";
                right = cur;
            }else if (ldist == rdist){
                if (hand == "left"){
                    answer += "L";
                    left = cur;
                }else{
                    answer += "R";
                    right = cur;
                }
            }else{
                answer += "L";
                left = cur;
            }
        }
    }
    return answer;
}