[프로그래머스] 코딩테스트 연습 > 코딩테스트 입문 > 문자열 밀기

예외 케이스를 혼자 힘으로 찾는다는 것은 정말 힘들지만 뿌듯한 일이다. 하마터면 못풀뻔했는데, 움직이지 않아도 되는 예외 케이스를 생각해보았다.

 

A가 "hello"이고, B 역시ㅣ "hello" 일 경우에, 움직이지 않아도 되기 때문에 답은 0이어야하지만, 실제로는 5번 움직여서 5로 답이 출력되는 것을 알 수 있었다. 그래서 답이 문자열의 길이와 같을 경우 0을 출력하도록 바꿨더니 정답이었다.

 

내가 맞힌 정답::

def solution(A, B):
    answer = 0
    
    n = len(A)
    
    for i in range(n):
        temp = ''
        temp = temp + A[-(i+1):] + A[:((n-1)-i)]
        print(temp)
        
        if temp == B:
            answer = i+1
            if answer == n:
                answer = 0
            break
        else :
            answer = -1
    
    return answer

https://school.programmers.co.kr/learn/courses/30/lessons/120921#

프로그래머스