프로그래머스 조이스틱 Swift

문제 설명

조이스틱으로 알파벳 이름을 완성하세요. 맨 처음엔 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~M까진 위로 올려야 더 작고 O~Z까지는 아래로 내려야 더 작다. (N은 가운데이므로 상관없다.)

그리고 좌우를 최소로 하는 방법은 

1.A의 가장 큰 집단이 왼쪽에 있는 경우(1번째 인덱스부터 ~) ex) BAAAAAABB

2.A의 가장 큰 집단이 오른쪽에 있는 경우 ex)BBAAAAA

3.A의 가장 큰 집단이 가운데에 있는 경우ex)BBAAAAAB

 

1의 경우 오른쪽으로 움직일 필요가 없으므로 왼쪽으로만 움작여서 A집단이 나오기 전까지 움직인 수와 각 알파벳의 상하 수를 구해서 더하면 된다.

2의 경우 왼쪽으로 움직이 필요가 없으므로 오른쪽으로만 움직여 A집단이 나오기 전까지 움직인 수와 각 알파벳의 상하 수를 구해서 더하면 된다.

3의 경우는 A집단 전까지와 후 중에 더 짧은 곳으로 결정된다. ex) BBAAAAABBBBBBB <- 전이 더 짧은 경우 BBBBBAAAAAAABB <- 후가 더 짧은 경우

3의 경우는 그냥 왼쪽이나 오른쪽으로 움직였을 때보다 더 짧은곳으로 갔다가 다시 되돌아와서 A집단 전까지 움직인 것과 비교를 해서 

더 작은 수가 알파벳의 상하 수와 더해지게 된다.

ex) BBBBBBBAAABBBBBBB <- A집단이 가운데에 있긴하지만 한 쪽으로 갔다가 되돌아오는 것보다 그냥 왼쪽이나 오른쪽으로 쭉 가는게 더 짧다.

ex) BBAAAAAAABB <- A집단이 가운데에 있고 A집단 전까지 오른쪽으로 이동했다가 다시 왼쪽으로 A집단 전까지 움직이면 한 쪽으로 쭉 가는 것보다 더 짧다.

func solution(_ name:String) -> Int {
    var answer = 0
    var start = 0
    var zero = [[Int]]()
    let n = name.utf8.map{$0-64}
    for i in 0..<n.count{
        if n[i] < 15 {
                answer += Int(n[i]) - 1
        }else{
            answer += 27 - Int(n[i])
        }
    }
    if name.contains("A"){
        if name == "A" {
            return 0
        }
    for i in 0..<n.count{
        if n[i] == 1 && i > 0{
            if start == 0 {
            start = i+1
            }
        }else{
            if start != 0 {
                zero.append([start,i])
                start = 0
            }else{
                continue
            }
        }
    }
    if start != 0 {
        zero.append([start,name.count])
    }
    let max = zero.map{$0.reduce(0){$1-$0}}
    for i in 0...max.count-1{
        if max[i] == max.max(){
            if zero[i][0] == 2 {
               answer += n.count - zero[i][1]
                break
            }else if zero[i][1] == n.count{
                answer += zero[i][0] - 2
 
            }else{
                let middleanswer = answer + name.count - 1
                if zero[i][0]-1 > name.count - zero[i][1]{
                    answer += (name.count - zero[i][1]) * 2 + zero[i][0] - 2
                }else{
                    answer += (zero[i][0] - 2) * 2 + name.count - zero[i][1]
                }
 
                if answer > middleanswer {
                    return middleanswer
                }
            }
 
        }
    }
        return answer
    }else{
        return answer + name.count-1
    }
}