Algorithm / 문자열 나누기

1) 문제

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

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

2) 문제 풀이

func solution(_ s:String) -> Int {
    var answer = 0
    var index = s.startIndex
    
    while index < s.endIndex {
        let x = s[index]
        var countX = 0
        var countNonX = 0
        
        var endIndexForGroup = index
        
        while endIndexForGroup < s.endIndex {
            let currentChar = s[endIndexForGroup]
            if currentChar == x {
                countX += 1
            } else {
                countNonX += 1
            }
            
            if countX == countNonX {
                endIndexForGroup = s.index(after: endIndexForGroup)
                break
            }
            
            endIndexForGroup = s.index(after: endIndexForGroup)
        }
        
        answer += 1
        index = endIndexForGroup
    }

    return answer
}

결과

3) 코드 개선

• index 관리 없이 for문을 사용하여 순회
• count가 0이 될 때마다 하나의 그룹을 완성
• 메모리 성능 및 가독성 향상

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

    for char in s {
        if x == nil {
            x = char
        }
        
        if char == x {
            countX += 1
        } else {
            countNotX += 1
        }
        
        if countX == countNotX {
            answer += 1
            x = nil
            countX = 0
            countNotX = 0
        }
    }
    
    // 남은 글자가 있을 경우 마지막 한 묶음 처리
    if x != nil {
        answer += 1
    }

    return answer
}

결과