본문 바로가기

알고리즘/프로그래머스

[ Python ] - 문자열 밀기(120921)

728x90
반응형

문제설명

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

제한사항

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

입출력 예

solution.py

def solution(A, B):
    answer = []  # A를 회전시켜 B와 같게 만들 수 있는 회전 횟수를 저장할 리스트

    if A == B:  # A와 B가 이미 같은 경우, 회전할 필요가 없으므로 0을 반환
        return 0
    
    for idx in range(len(A)):  # A의 길이만큼 반복하여 모든 가능한 회전을 고려
        # A 문자열의 뒷부분을 앞으로, 앞부분을 뒤로 회전시켜 B와 같은지 비교
        if A[-idx:] + A[:-idx] == B:  # 만약 회전시킨 결과가 B와 같다면
            answer.append(idx)  # 해당 회전 횟수를 answer 리스트에 추가
    
    # answer 리스트에 가능한 회전 횟수가 있다면, 그 중 최소값을 반환
    # 가능한 회전 횟수가 없다면 -1을 반환
    return min(answer) if answer else -1
728x90
반응형