[프로그래머스] LV.0 문자열 밀기

문제

문자열 "hello"에서 각 문자를 오른쪽으로 한 칸씩 밀고 마지막 문자는 맨 앞으로 이동시키면 "ohell"이 됩니다. 이것을 문자열을 민다고 정의한다면 문자열 A와 B가 매개변수로 주어질 때, A를 밀어서 B가 될 수 있다면 밀어야 하는 최소 횟수를 return하고 밀어서 B가 될 수 없으면 -1을 return 하도록 solution 함수를 완성해보세요.

제한사항

- 0 < A의 길이 = B의 길이 < 100
- A, B는 알파벳 소문자로 이루어져 있습니다.

입출력 예시

A	B	result
"hello"	"ohell"	1
"apple"	"elppa"	-1
"atat"	"tata"	1
"abc"	"abc"	0

풀이

1. 문자열의 제일 뒤에있는 문자를 맨 앞으로 가져오는 알고리즘을 구성해야한다.
2. 만약 그렇게 변경한 A가 B와 같지 않으면 계속해서 맨 뒤 문자를 맨 앞으로 가져와야하는데 이 때 한 바퀴 돌게되면 멈추고 A와 B가 같아질 수 없다는 return 값을 반환한다.
3. 또한 A와 B가 애초에 같을 경우 또한 생각해주도록 한다.

풀이

def solution(A, B):
    cnt = 0
    if A == B:
        return 0
    else:
        while cnt <= len(A)-1:
            # -1 을 맨 앞으로 보내고 문자 만들기
            cnt+=1
            new_A = ''
            new_A+=A[len(A)-1]
            for a in range(0,len(A)-1):
                new_A+=A[a]
            A = new_A
            if A == B:
                break
    if cnt == len(A):
        return -1
    return cnt

• 먼저 맨 뒤 문자를 맨 앞으로 보내는 회수를 카운트해주기 위해 cnt변수를 0으로 초기화해준다.
• if문으로 A와 B가 애초에 같을 때 0을 return 해주고 아닐 때는 맨 뒤 문자를 맨 앞으로 보내주는 과정을 거쳐준다.
• A와 B가 다를 때 while문으로 A의 길이-1 만큼 반복시켜주기 위해 cnt<=len(A)-1을 지정해주고 cnt를 매 반복마다 1씩 증가시켜준다. 그리고 새로 생성될 new_A를 매 반복마다 초기화시켜주고 A의 마지막 값을 new_A의 처음에 넣어주고 반복문을 통해 나머지 문자들을 넣어준다. 이후 A를 new_A의 값으로 변경시켜 다음 반복문이 돌 때 A 값이 변경되게끔 해준다. 이 때 if문으로 변경된 A의 값과 B의 값이 같아지는 경우가 있으면 while문에서 벗어나도록 break 해준다.
• 이 때 cnt의 값이 len(A)의 값과 같아지면 while문이 한 바퀴 돈 것이므로 A와 B가 같을 확률이 없으므로 -1을 return 해준다.
• 정상적으로 같을 경우에는 cnt 값을 반환하여 몇 번 자리를 바꿔줬는데 return 해준다.

오답노트

• 문자를 [순서]를 통해 따로따로 사용할 수 있다.
• range()를 좀 더 자세히 공부할 필요가 있다.