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

https://school.programmers.co.kr/learn/courses/30/lessons/140108

프로그래머스

 

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

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

s에서 분리한 문자열을 빼고 남은 부분에 대해서 이 과정을 반복합니다. 남은 부분이 없다면 종료합니다.
만약 두 횟수가 다른 상태에서 더 이상 읽을 글자가 없다면, 역시 지금까지 읽은 문자열을 분리하고, 종료합니다.

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

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

입출력 예
s	result
"banana"	3
"abracadabra"	6
"aaabbaccccabba"	3

 

문제를 풀기는 했는데 조금 지저분하게 풀어서 아쉬웠다.

function solution(s) {
    let answer = [];
    let x = s[0];
    let num = 1;
    let stack = [x];
    
    for(let i = 1; i < s.length; i++) {
        if(x === s[i]) {
            num++;
        } else {
            num--;
        }
        stack.push(s[i]);
        
        if(num === 0) {
            answer.push(stack.join(''));
            stack = [];
            
            if(i + 1 <= s.length) {
                x = s[i + 1];
            } else {
                break;
            }
        }
    }
    
    return answer.join('').length === s.length ? answer.length : answer.length + 1;
}

 

문제 그래도를 구현한 느낌이라 더 아쉬운 거 같다. 효율적이고 아름다운 코드는 아래에 있다.

 

function solution(s) {
    let answer = 0;
    let current;
    let count = 0;

    for(let i = 0; i < s.length; i++) {
        if(count === 0) {
            answer++;
            current = s[i]
            count = 1
        } else {
            if(current !== s[i]) count--;
            else count++;
        }
    }

    return answer;
}