처음 문제를 보고 되게 간단할 줄 알았으나 일부 접근 방법에 대해서 정확하게 생각하기 어려운 부분이 있었다.
우선 문제를 살펴보자.
문제
조이스틱으로 알파벳 이름을 완성하세요. 맨 처음엔 A로만 이루어져 있습니다.
ex) 완성해야 하는 이름이 세 글자면 AAA, 네 글자면 AAAA
조이스틱을 각 방향으로 움직이면 아래와 같습니다.
▲ - 다음 알파벳
▼ - 이전 알파벳 (A에서 아래쪽으로 이동하면 Z로)
◀ - 커서를 왼쪽으로 이동 (첫 번째 위치에서 왼쪽으로 이동하면 마지막 문자에 커서)
▶ - 커서를 오른쪽으로 이동 (마지막 위치에서 오른쪽으로 이동하면 첫 번째 문자에 커서)
예를 들어 아래의 방법으로 "JAZ"를 만들 수 있습니다.
- 첫 번째 위치에서 조이스틱을 위로 9번 조작하여 J를 완성합니다.
- 조이스틱을 왼쪽으로 1번 조작하여 커서를 마지막 문자 위치로 이동시킵니다.
- 마지막 위치에서 조이스틱을 아래로 1번 조작하여 Z를 완성합니다.
따라서 11번 이동시켜 "JAZ"를 만들 수 있고, 이때가 최소 이동입니다.
만들고자 하는 이름 name이 매개변수로 주어질 때, 이름에 대해 조이스틱 조작 횟수의 최솟값을 return 하도록 solution 함수를 만드세요.
제한 사항
- name은 알파벳 대문자로만 이루어져 있습니다.
- name의 길이는 1 이상 20 이하입니다.
입출력 예
| name | return |
|---|---|
| "JEROEN" | 56 |
| "JAN" | 23 |
접근 방법
-
조이스틱의 상하 조작 케이스
처음 'A'부터
위로 조작해 알파벳을 찾는 경우와
처음 'A'부터아래로 조작해 알파벳을 찾는 경우가 있다.
이 케이스는 단순히 for 문을 돌면서 두 경우 중 작은 값을 추가해주면 된다. -
조이스틱의 좌우 조작 케이스
좌우 조작 케이스는 총 3가지가 있다.
첫 번째는 처음부터 멈추지 않고 오른쪽 끝까지 가는 경우 (name.length - 1)
두 번째는 처음부터 오른쪽으로 갔다가 'A'를 만나면 다시 처음으로 돌아와서 왼쪽으로 가는 경우
ex) BBAAAAAACCC 의 경우, BB까지 갔다가 다시 돌아와서 왼쪽으로 더 가서 CCC를 완성한다.
세 번째는 처음부터 왼쪽으로 갔다가 'A'를 만나면 다시 처음으로 돌아와서 오른쪽으로 가는 경우 (2 + 3 = 5)
ex) CCCAAAAAAY 의 경우, 처음부터 왼쪽으로 움직여서 Y를 완성해주고 다시 오른쪽으로 움직이면서 CCC를 완성해준다. (2 + 2 = 4)
결국 이 문제의 핵심은 좌우 커서 이동 값의 최솟값을 찾아주는 것이 핵심이였다.
코드
import kotlin.math.*
class Solution {
fun solution(name: String): Int {
var vertical = 0
var horizontal = name.length - 1 // 처음부터 오른쪽 끝까지 가는 경우
var index = 0
for (i in 0 until name.length) {
vertical += min(name[i] - 'A', 'Z' - name[i] + 1)
// 연속되는 A 갯수 확인
index = i + 1
while (index < name.length && name[index] == 'A') {
index++
}
horizontal = min(horizontal, i * 2 + name.length - index) // 오른쪽 갔다가 돌아와서 쭉
horizontal = min(horizontal, (name.length - index) * 2 + i) // 왼쪽 갔다가 돌아와서 쭉
}
return vertical + horizontal
}
}