문자열 밀기

2024. 9. 27. 18:00코딩테스트 입문

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


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


제한사항

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

입출력 예

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

입출력 예 설명
입출력 예 #1

  • "hello"를 오른쪽으로 한 칸 밀면 "ohell"가 됩니다.

입출력 예 #2

  • "apple"은 몇 번을 밀어도 "elppa"가 될 수 없습니다.

입출력 예 #3

  • "atat"는 오른쪽으로 한 칸, 세 칸을 밀면 "tata"가 되므로 최소 횟수인 1을 반환합니다.

입출력 예 #4

  • "abc"는 밀지 않아도 "abc"이므로 0을 반환합니다.

문자열 A를 오른쪽으로 회전시켜서 문자열 B와 같아지는 시점을 찾는 문제입니다.

 

문자열을 회전시킨다는 건 겉보기는 조금 어려울 수 있는데, k개를 잘라서 뒤로 붙이면 회전이랑 동일하다는 걸 알 수 있습니다.

이 문제에서는 오른쪽으로 회전시키기 때문에, 음수 인덱싱을 사용해서 풀면 됩니다.

파이썬으로 문제를 풀어보겠습니다.

def solution(A, B):
    # A의 길이만큼 회전시킵니다.
    for k in range(len(A)):
        # 오른쪽으로 회전시킵니다.
        if A[-k:] + A[:-k] == B: # B랑 같아지면 k를 반환합니다.
            return k
    # 일치하지 않는다면 -1 반환
    return -1

 

A 문자열의 뒷 k개를 잘라서 앞 문자열에 붙이는 식으로 오른쪽 회전을 구현했습니다.

B랑 같아지는 시점에 k를 반환하는데, 만약 return이 실행되지 않고 반복문이 끝난다면 회전을 해도 일치하는 경우가 없다는 뜻이므로 -1을 반환합니다.

 

슬라이싱을 활용해서 회전을 구현할 수 있다는 것만 기억하시면 될 것 같습니다.

'코딩테스트 입문' 카테고리의 다른 글

연속된 수의 합  (0) 2024.09.29
종이 자르기  (0) 2024.09.28
잘라서 배열로 저장하기  (1) 2024.09.26
7의 개수  (0) 2024.09.25
문자열 정렬하기 (2)  (0) 2024.09.24