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

프로그래머스 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

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

입출력 예 #2

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

입출력 예 #3

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

입출력 예 #4

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

---

제출

import Foundation

func solution(_ A:String, _ B:String) -> Int {
    if A == B {return 0}
    var sValue = A
    for i in 1..<sValue.count{
        var temp = sValue.removeLast()
        sValue.insert(temp, at: sValue.startIndex)
        if sValue == B {return i}
    }

    return -1
}

두 문자열이 동일하다면 0을 반환한다.

i를 1부터 문자열의 길이까지 증가시키며 반복문을 실행한다.
A의 마지막 값을 삭제한 후 맨 앞에 추가했을 때, B와 동일하다면 i를 반환한다. (문자열을 오른쪽으로 미는 동작)

for문이 끝날 때까지 B와 같아지는 경우가 없다면, A는 B가 될 수 없기 때문에 -1을 반환한다. 

다른 풀이

import Foundation

func solution(_ A:String, _ B:String) -> Int {
    let arr = (A+A).components(separatedBy: B)
    return arr.count == 1 ? -1 : arr[1].count
}

A에 A를 더해준 후, B를 기준으로 나눈다.
배열의 크기가 1이라면, A는 B가 될 수 없다는 의미이기 때문에 -1을 반환한다.
배열의 크기가 1이 아니라면, 배열의 첫 번째 요소의 크기를 반환한다.