[프로그래머스/Java] Lv.0 코드 처리하기

문제

문자열 code가 주어집니다.
code를 앞에서부터 읽으면서 만약 문자가 "1"이면 mode를 바꿉니다. mode에 따라 code를 읽어가면서 문자열 ret을 만들어냅니다.

mode는 0과 1이 있으며, idx를 0 부터 code의 길이 - 1 까지 1씩 키워나가면서 code[idx]의 값에 따라 다음과 같이 행동합니다.

• mode가 0일 때
  code[idx]가 "1"이 아니면 idx가 짝수일 때만 ret의 맨 뒤에 code[idx]를 추가합니다.
  code[idx]가 "1"이면 mode를 0에서 1로 바꿉니다.
• mode가 1일 때
  code[idx]가 "1"이 아니면 idx가 홀수일 때만 ret의 맨 뒤에 code[idx]를 추가합니다.
  code[idx]가 "1"이면 mode를 1에서 0으로 바꿉니다.

문자열 code를 통해 만들어진 문자열 ret를 return 하는 solution 함수를 완성해 주세요.
단, 시작할 때 mode는 0이며, return 하려는 ret가 만약 빈 문자열이라면 대신 "EMPTY"를 return 합니다.

- 제한사항

• 1 ≤ code의 길이 ≤ 100,000
• code는 알파벳 소문자 또는 "1"로 이루어진 문자열입니다.

풀이

- 오답

문제를 읽자마자 mode가 토글 형태로 동작한다는 생각이 들었고, 변경 사항을 code[idx] 값에 따라 조절할 수 있도록 변수 int mode를 선언했다.

아래와 같이 코드를 작성하고 보니 지저분하기도 하거니와 '실행한 결괏값 "ac1bac"이 기댓값 "acbac"과 다릅니다.'라는 오류가 확인되었다.

해결하는 김에 코드를 다듬어 보기로 했다.

class Solution {
    public String solution(String code) {
        String answer = "";
        int mode = 0;
        
        for(int i=0;i<code.length();i++){
            if(code.charAt(i) == '1') mode = 1 - mode;
            
            if(mode == 0 && i%2 == 0){
                answer += code.charAt(i);
            }else if(mode == 1 && i%2 == 1){
                answer += code.charAt(i);
            }
        }
        
        return answer.equals("") ? "EMPTY" : answer;
        
    }
}

- 정답

class Solution {
    public String solution(String code) {
        String answer = "";
        int mode = 0;
        
        for(int i=0;i<code.length();i++){
            if(code.charAt(i) == '1') mode = 1 - mode;
            else if(i%2 == mode) answer += code.charAt(i);
        }
        
        return answer.equals("") ? "EMPTY" : answer;
        
    }
}

결과