[Swift] 프로그래머스 LV.1 문자열 나누기

프로그래머스 LV.1 모음

문자열 나누기

문제 설명

문자열 s가 입력되었을 때 다음 규칙을 따라서 이 문자열을 여러 문자열로 분해하려고 합니다.

• 먼저 첫 글자를 읽습니다. 이 글자를 x라고 합시다.
• 이제 이 문자열을 왼쪽에서 오른쪽으로 읽어나가면서, x와 x가 아닌 다른 글자들이 나온 횟수를 각각 셉니다. 처음으로 두 횟수가 같아지는 순간 멈추고, 지금까지 읽은 문자열을 분리합니다.
• s에서 분리한 문자열을 빼고 남은 부분에 대해서 이 과정을 반복합니다. 남은 부분이 없다면 종료합니다.
• 만약 두 횟수가 다른 상태에서 더 이상 읽을 글자가 없다면, 역시 지금까지 읽은 문자열을 분리하고, 종료합니다.

문자열 s가 매개변수로 주어질 때, 위 과정과 같이 문자열들로 분해하고, 분해한 문자열의 개수를 return 하는 함수 solution을 완성하세요.

제한 사항

• 1 ≤ s의 길이 ≤ 10,000
• s는 영어 소문자로만 이루어져 있습니다.

입출력 예

s	result
"banana"	3
"abracadabra"	6
"aaabbaccccabba"	3

입출력 예 설명

입출력 예 #1

• s = "banana"인 경우 ba - na - na와 같이 분해됩니다.

입출력 예 #2

• s = "abracadabra"인 경우 ab - ra - ca - da - br - a와 같이 분해됩니다.

입출력 예 #3

• s = "aaabbaccccabba"인 경우 aaabbacc - ccab - ba와 같이 분해됩니다.

---

제출

import Foundation

func solution(_ s:String) -> Int {
    var stringValue = s.map{String($0)}
    var x = stringValue[0], x_cnt = 0, not_x_cnt = 0
    var result = 0
    
    for i in stringValue{
        if x_cnt != 0 && x_cnt == not_x_cnt{
            result += 1
            x_cnt = 0
            not_x_cnt = 0
            x = i
        }
        
        if x == i {x_cnt += 1}
        else {not_x_cnt += 1}
    }
    
    return result+1
}

입력받은 문자열 s를 배열로 변환해 stringValue에 저장한다.
기준이 되는 문자를 담을 변수 x와, x와 x가 아닌 다른 글자의 개수를 저장할 변수  x_cnt, not_x_cnt를 생성한다.

stringValue 배열의 요소를 순서대로 하나씩 탐색한다.
x의 개수가 0이 아니고, x의 개수와 x가 아닌 다른 글자의 개수가 같을 때, result 변수를 1 증가시킨 후 x_cnt와 not_x_cnt 모두 0으로 초기화한다. 또한 기준이 되는 문자 x를 i로 바꾼다.
x가 i와 같으면 x_cnt를 1 증가시키고, x와 i가 다르다면 not_x_cnt를 증가시킨다.

마지막 분해한 문자열을 result 변수에 더해지지 않기 때문에, result+1을 반환한다.