본문 바로가기

⏳ 알고리즘/python 알고리즘 문제 풀이

프로그래머스 - LV2. 조이스틱

문제

programmers.co.kr/learn/courses/30/lessons/42860#

 

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

조이스틱으로 알파벳 이름을 완성하세요. 맨 처음엔 A로만 이루어져 있습니다. ex) 완성해야 하는 이름이 세 글자면 AAA, 네 글자면 AAAA 조이스틱을 각 방향으로 움직이면 아래와 같습니다. ▲ - 다

programmers.co.kr

 

코드

def solution(name):
    arr = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
    origin = ["A"]*len(name)
    count = [0]*len(name)
    cnt = 0
    
    if len(name) == 1:
        if arr.index(name[0]) >= 13:
            return 26-arr.index(name)
        else:
            return arr.index(name)
    
    #상하 조정
    i = 0
    while origin != name and i<len(name):
        cnt += 1
        if arr.index(name[i]) >= 13:
            origin[i] = name[i]
            count[i] = 26-arr.index(name[i])
        else:
            origin[i] = name[i]
            count[i] = arr.index(name[i])
        i += 1
        
    #좌우 조정    
    result = sum(count)
    j = 0
    while count != [0]*len(name):
        count[j] = 0
        if count == [0]*len(name):
            break
        else:
            result += 1
        if name[1] == "A" and name[-1] != "A":
            j -= 1
        else:
            j += 1
        
    return result

테스트 11이 실패했다. 테스트 11은 무슨 경우일까?