문자열 "hello"에서 각 문자를 오른쪽으로 한 칸씩 밀고 마지막 문자는 맨 앞으로 이동시키면 "ohell"이 됩니다. 이것을 문자열을 민다고 정의한다면 문자열 A와 B가 매개변수로 주어질 때, A를 밀어서 B가 될 수 있다면 밀어야 하는 최소 횟수를 return하고 밀어서 B가 될 수 없으면 -1을 return 하도록 solution 함수를 완성해보세요.
A | B | result |
---|---|---|
"hello" | "ohell" | 1 |
"apple" | "elppa" | -1 |
"atat" | "tata" | 1 |
"abc" | "abc" | 0 |
입출력 예 설명
1. 탈출조건 1) A == B: return cnt
2. 탈출조건 2) B를 만들 수 없다면: return -1
3. 반복) A = 문자열 1 칸 밀기 cnt += 1
# 문자열 밀면서 확인하는 재귀 함수
def valid(a, b, cnt=0):
if a == b:
return cnt
if cnt == len(a): # B를 만들 수 없다면
return -1
return valid(a[-1]+a[:-1], b, cnt+1)
def solution(A, B):
return valid(A, B)
find
함수를 통해 생성 가능한 단어인지 확인한다.find
함수가 찾고자 하는 문자열이 없을 때 -1을 return한다.lambda
를 사용해 더 간단하게 함수를 작성했다.solution = lambda a,b:(b*2).find(a)
rotate
함수를 활용해서 문자열을 밀면서 문제를 해결할 수 있다.from collections import deque
def solution(A, B):
a, b = deque(A), deque(B)
for cnt in range(len(A)):
if a == b:
return cnt
a.rotate(1)
return -1