[Level 2] 조이스틱

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

 

def cnt_up_down(chr):
    cnt_from_a = abs(ord('A')-ord(chr))
    cnt_from_z = abs(ord('Z')-ord(chr)+1)
    
    return(min(cnt_from_a, cnt_from_z))

def solution(name):
    cnt = 0

    for i in name:
        cnt += cnt_up_down(i)

    lst_name = list(name)
    len_name = len(lst_name)
    
    
    if lst_name.count('A') == len_name:
        return 0

    cur = 0
    lst_name[cur] = 'A'
    

    while lst_name.count('A') != len_name:
        cnt_left = 0
        cnt_right = 0
    
        pos_right, pos_left = cur, cur

        for i in range(1, len_name):
            cnt_right += 1
            pos_right += 1
        
            if pos_right > len_name-1:
                pos_right -= len_name
            
            if lst_name[pos_right] != 'A':
                break
            
        for j in range(1, len_name):
            cnt_left += 1
            pos_left -= 1
        
            if pos_left < 0:
                pos_left += len_name
            
            if lst_name[pos_left] != 'A':
                break
            
        if cnt_right <= cnt_left:
            lst_name[pos_right] = 'A'
            cnt += cnt_right
            cur = pos_right
        else:
            lst_name[pos_left] = 'A'
            cnt += cnt_left
            cur = pos_left

    return cnt

 

▷ 이 문제는 크게 조이스틱을 위/아래로 움직이는 경우와 좌/우로 움직이는 경우로 나누어 풀어야 한다. 위/아래로 움직이는 횟수는 어떠한 경우에도 같다. 따라서 좌/우의 이동을 줄여서 최소의 이동횟수로 만드는 것이 핵심이다.

 

▷ cnt_up_down 함수는 문자에 대한 위/아래로 움직여야 하는 횟수를 구하는 함수이다.

 

▷ 위 코드의 논리를 요약하면 다음과 같다.

 

1. cnt_up_down 함수를 이용하여 위/아래로 움직이는 횟수를 구한다.

2. 첫 번째 위치의 문자를 A로 변경한다.

3. 첫 번째 위치를 현재 위치로 정한 뒤, 왼쪽으로 이동하는 경우와 오른쪽으로 이동하는 경우의 이동횟수를 비교한다. 이때 둘 중 작은 이동횟수를 전체 이동횟수에 추가하고, 현재 위치를 해당 이동을 반영하여 변경한다. 이때 변경된 현재위치의 문자를 A로 변경한다.

4. name 변수의 모든 문자가 A가 될 때까지 3번 작업을 반복한다.