문자열 밀기
2024. 9. 27. 18:00ㆍ코딩테스트 입문
https://school.programmers.co.kr/learn/courses/30/lessons/120921
문제 설명
문자열 "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
- "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 |