[프로그래머스] 21. 코드 처리하기

입출력 예

입출력 예 설명

• code의 각 인덱스 i에 따라 다음과 같이 mode와 ret가 변합니다.

코드

class Solution {
    public String solution(String code) {
        char[] charArr = code.toCharArray();
        String ret = "";
        int mode = 0;
        
        for (int i = 0; i < charArr.length; i++) {
            if(mode == 0) {
                if(charArr[i] == '1') {
                    mode = 1;
                } else {
                    if(i % 2 == 0) ret += charArr[i];
                }
            } else {
                if(charArr[i] == '1') {
                    mode = 0;
                } else {
                    if(i % 2 != 0) ret += charArr[i];
                }
            }
        }
        
        return ret.length() < 1 ? "EMPTY" : ret;
    }
}

• if(charArr[i] == '1')
• 이 부분에서 숫자 1이랑 비교를 한다고 계속 실패함
• char 타입 ‘1’과 비교를 해야함

더 좋은 코드

class Solution {
    public String solution(String code) {
        StringBuilder answer = new StringBuilder();
        int mode = 0;
        for (int i = 0; i < code.length(); i++) {
            char current = code.charAt(i);
            if (current == '1') {
                mode = mode == 0 ? 1 : 0;
                continue;
            }

            if (i % 2 == mode) {
                answer.append(current);
            }
        }
        return answer.length() == 0 ? "EMPTY" : answer.toString();
    }
}

• StringBuilder를 사용해서 가변 문자열을 사용한다.
• 매번 새로운 문자열을 만들지 않는다
• 내부적으로 char[]을 사용한다

• 모드값에 따른 처리 과정을 먼저 수행하고

• 모드값이 1 or 0 이니 이 값을 가지고 홀수 짝수 비교를 한다