[ 프로그래머스 ] 문자열 나누기

 

 

 

🍄 코딩테스트

코딩테스트 문제 풀이

✍🏻 Github

문제 풀이 github 링크

 

 

문제 설명

문자열 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 stack = [Character]()
    var count = 0
    var answer = 0
    
    for char in s {
        if stack == [] {
            stack.append(char)
        }else if stack.last == char {
            stack.append(char)
        }else {
            count += 1
        }
        
        if stack.count == count {
            stack = []
            answer += 1
            count = 0 
        }
    }
    if stack != [] {
        answer += 1 
    }
    return answer
}

 

배열을 사용하여 풀었다.

배열에 아무것도 들어있지 않다면, 배열에 추가하거나 배열 마지막에 든 원소와 같다면 배열에 추가하도록 하였다.

x의 개수는 stack.count로 파악하고, x가 아닌 친구들의 개수는 count를 1씩 높이면서 파악하였다.

그렇게 stack.count와 count가 같아진다면, 문자열이 하나 분리가 되어 answer을 1을 높여주었다.

 

문제에서 문자열을 분리하여 출력하는 문제가 아니기 때문에 문자열에 신경을 쓰는 게 아닌 개수에 초점을 두어 풀었다.

 

 

" 만약 두 횟수가 다른 상태에서 더 이상 읽을 글자가 없다면 지금까지 읽은 문자열을 분리한다"

 

라는 조건이 있어서 반복문을 끝나고 난 다음에도 stack이 비어있지 않다면 위의 경우에 해당하는 경우이므로 answer 개수를 하나 더 증가시켰다.

 

이 경우엔 s의 길이가 최대 10,000이므로 시간 복잡도 O(n)이 최대일 것이다.

 

 

다른 사람 풀이

import Foundation

func solution(_ s:String) -> Int {
    var answer = 0
    var x: Character? = nil
    var xCount = 0

    for i in s {
        if x == nil { 
            x = i 
            xCount = 1
            answer += 1
            continue
        }

        xCount += x == i ? 1 : -1

        if xCount == 0 {
            x = nil
        }
    }

    return answer
}

 

배열로 풀지 않고 문자열로 풀었다.

배열보다 시간을 더 줄일 수 있을 것 같아 좋은 방법이라고 생각했다.