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

문제

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

프로그래머스

문제 설명

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

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

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

 

제한사항

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

 

풀이

• 분해한 문자열들을 저장할 리스트 생성
• while 루프를 통해 s가 빈 문자열이 될 때까지 반복
  • x 라는 변수에 s의 첫 글자를 저장
  • countX 와 countNonX 에 x와 x가 아닌 다른 글자들이 나온 횟수를 저장
  • for 루프를 통해  countX와 countNonX가 같아질 때까지 반복
  • 같아지면 substrings에 추가, 그리고 분리한 부분 이후의 문자열을 s에 저장
  • 같지 않은데 읽을 글자가 없으면 반복문 종료
• 리스트 크기 반환

import java.util.*;

class Solution {
    public int solution(String s) {
        List<String> substrings = new ArrayList<>();
        
        while (!s.isEmpty()) {
            char x = s.charAt(0);
            int countX = 1;
            int countNonX = 0;
            
            for (int i = 1; i < s.length(); i++) {
                char currentChar = s.charAt(i);
                if (currentChar == x) {
                    countX++;
                } else {
                    countNonX++;
                }
                
                if (countX == countNonX) {
                    substrings.add(s.substring(0, i + 1));
                    s = s.substring(i + 1);
                    break;
                }
            }
            
            if (countX != countNonX) {
                substrings.add(s);
                break;
            }
        }
        
        return substrings.size();
    }
}