[프로그래머스 / C++] 조이스틱

 

문제링크: programmers.co.kr/learn/courses/30/lessons/42860

코딩테스트 연습 - 조이스틱

 

풀이

 

조이스틱을 위아래로 움직이는 횟수와 좌우로 움직이는 횟수를 따로 생각했다. 

 

위아래로 움직이는 것은 알파벳에서 A가 가까운지 Z가 가까운지에 따라 최소 횟수를 구했다. 

 

좌우로 움직이는 것은 왼쪽과 오른쪽 중 어느 쪽으로 가야 최소가 되는지를 구해서 해결했다. 좌우 이동 횟수(move)의 초깃값은 최대 이동 횟수인 len - 1이다. 이후 i + (len - next) + min(i, len - next)과 비교해 최솟값을 구한다. 

 

ABBAAAABA를 예시로 계산을 설명하자면, i가 2일 때 next는 7이 된다. i에서 next까지 왼쪽으로 이동했을 때의 거리는 i + (len - next)이다. 처음에 왼쪽으로 갈지 오른쪽으로 갈지는 min(i, len - next)로 구한다. 이 예시에선 len - next의 값이 더 작으므로 왼쪽으로 먼저 가는 것을 택한다. 따라서 len - next를 i와 next의 거리에 더해주어 왼쪽으로 갔다가 돌아오는 것을 계산해준다. 

 

 

 

소스코드

 

#include <string>
#include <vector>
#include <algorithm>
 
using namespace std;
 
int solution(string name) {
    int answer = 0; 
    for(auto ch: name) answer += min(ch - 'A', 91 - ch);
 
    int len = name.length();
    int move = len - 1;
    int next;
    for(int i = 0; i < len; i++) {
        next = i + 1;
        while(next < len && name[next] == 'A') next++;
        move = min(move, i + (len - next) + min(i, len - next));
    }
    answer += move;
 
    return answer;
}

테스트 결과

 

 

공부한 것을 정리한 내용입니다. 수정할 부분이 있다면 알려주시면 감사하겠습니다 :)