[프로그래머스 2020 카카오 인턴십] 키패드 누르기

문제

programmers.co.kr/learn/courses/30/lessons/67256

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

 

풀이

$$(R, C) = ((numbers[i] - 1) / 3, (numbers[i] - 1) \% 3)$$

1. 키패드를 좌표계라고 생각한다면 0을 제외한 번호 i의 좌표는 위와 같이 바꿀 수 있고, 마찬가지로 손의 위치도 간단하게 좌표로 나타낼 수 있습니다. (0번의 좌표 (R, C) = (3, 1))

 

2. 양 사이드의 번호를 누를 때에는 가까운 위치의 손을 사용하면 되므로 중간에 있는 2, 5, 8, 0을 어떤 위치의 손으로 누를지만 양쪽 손과의 거리를 비교하고 정해주면 됩니다.

 

코드

#include <bits/stdc++.h>
using namespace std;

pair<int, int> l, r;
int R, C;
int d(int r1, int c1, int r2, int c2) { return (abs(r2 - r1) + abs(c2 - c1)); }

string solution(vector<int> numbers, string hand) {
	string answer = "";
	l = { 3,0 }, r = { 3,2 };
	for (int i = 0; i < numbers.size(); i++) {
		bool left = 0;
		// #1. 키패드번호를 좌표로 변환
		if (!numbers[i]) R = 3, C = 1;
		else R = (numbers[i] - 1) / 3, C = (numbers[i] - 1) % 3; 

		// #2. 사이드쪽은 가까운 위치의 손을 사용, 가운데는 양쪽 손과 거리비교
		if (numbers[i] % 3 == 1) left = 1; 
		else if (numbers[i] == 0 || numbers[i] % 3 == 2) {
			int d1 = d(l.first, l.second, R, C), d2 = d(r.first, r.second, R, C);
			if (d1 < d2 || ((d1 == d2) && hand == "left")) left = 1;
		}
		if (left) answer += 'L', l = { R,C };
		else answer += 'R', r = { R,C };	
	}
	return answer;
}