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

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

먼저 첫 글자를 읽습니다. 이 글자를 x라고 합시다.
이제 이 문자열을 왼쪽에서 오른쪽으로 읽어나가면서, x와 x가 아닌 다른 글자들이 나온 횟수를 각각 셉니다. 처음으로 두 횟수가 같아지는 순간 멈추고, 지금까지 읽은 문자열을 분리합니다.

class Solution {
    fun solution(s: String): Int {
        var answer: Int = 0

        var chr = ' '
        var cnt = 0
        for (c in s) {
            when (chr) {
                ' ', c -> {
                    cnt++
                    chr = c
                }
                else -> {
                    cnt--
                }
            }

            if (cnt == 0) {
                answer++
                chr = ' '
            }
        }

        if (cnt > 0) answer++

        return answer
    }
}