[programmers] 프로그래머스 조이스틱(Python)

(주)그렙

[programmers] 프로그래머스 조이스틱

(Python)

(글쓴날 : 2020.06.19)

 

---

* programmers, 프로그래머스 문제 Python 언어 풀이입니다.

* 소스 코드의 저작권은 글쓴이에게 있습니다.

---

 

 

프로그래머스 조이스틱

---

1) 문제

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

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

---

2) 풀이 과정

* 시간 복잡도 : O(n)

 

영어 대문자로 이루어진 문자열이 주어질 때, 특정 규칙에 맞춰 모두 'A'로 변환하는 최소 횟수를 구하는 문제입니다.

 

저의 경우, 그리디를 적용하였고, Python을 사용했습니다.

먼저, 주어지는 문자열을 리스트로 변환한 후, 'A'가 아닌 문자의 개수를 세어 모두 'A'로 이루어진 문자열을 예외 처리함과 동시에 반복문을 탈출할 조건을 만들었습니다.

그 후, 'A'가 아닌 문자를 만날 때마다 위아래 방향으로 최소 횟수만큼 움직여 'A'로 바꾼 뒤, 다음 'A'가 아닌 문자를 찾는데, 현재 위치에서 오른쪽, 왼쪽으로 탐색해가며 두 방향 중 최소 횟수인 곳을 먼저 찾았습니다. 만약, 두 방향의 횟수가 서로 같다면 오른쪽에 우선순위가 있으므로 오른쪽을 체크하는 조건문을 왼쪽을 체크하는 조건문보다 위에 두었고, 그렇게 총 움직인 횟수를 구해 문제를 해결했습니다.

---

3) 코드

 

* Python 코드

def solution(name):
    answer = 0
    name = list(name)
    
    shouldChange = 0
    for n in name:
        if n != 'A':
            shouldChange += 1
    
    if not shouldChange:
        return 0
    
    goLeft = 0
    goRight = 0
    index = 0
    
    while 1:
        if name[index] != 'A':
            answer += min(ord('Z') - ord(name[index]) + 1, ord(name[index]) - ord('A'))
            
            name[index] = 'A'
            shouldChange -= 1
        
        if not shouldChange:
            break
            
        goLeft = goRight = index
        move = 0
 
        while 1:
            goRight += 1
            goLeft -= 1
            move += 1
 
            if goLeft < 0:
                goLeft = len(name) - 1
            if goRight > len(name) - 1:
                goRight = 0
 
            if name[goRight] != 'A':
                answer += move
                index = goRight
                break
            elif name[goLeft] != 'A':
                answer += move
                index = goLeft
                break
            
    return answer

---