[Java] 코드 처리하기

문제

• 문자열 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"로 이루어진 문자열입니다.

입출력 예

문제 풀이

맨 처음에 문제를 읽는데 이해가 안 가서 애를 먹었다.

문자열은 쌍따옴표로 감싸지만, 문자는 홑따옴표로 감싼다.
그래서 charAt()를 사용해서 문자로 변환한 다음 a != '1' 로 작성했다.
substring()을 사용해서 잘랐다면 !a.equals("1") 로 작성해야한다.

처음 작성한 코드는 아래와 같다.

class Solution {
    public String solution(String code) {
        String answer = "";
        int mode = 0;
        
        for (int i = 0; i < code.length(); i++) {
            char a = code.charAt(i);
        
            if (mode == 0) {
                if (a != '1') {
                    if (i % 2 == 0) {
                    answer += a;
                    }
                } else {
                    mode = 1;
                }

            } else if (mode == 1) {
                if (a != '1') {
                    if (i % 2 != 0) {
                    answer += a;
                    }
                } else {
                    mode = 0;
                }
            } 
        }
        if (answer.equals("")) {
            answer = "EMPTY";
        }
        
        return answer;
    }
}

다른 사람들과 코드를 비교해 보니 제법 길다. 참고하여 어떻게 줄일 수 있을까 고민해보았다.

a가 1이 아닌 경우도 (i % 2 == mode) 라고 쓰면 i가 홀수일 때 mode는 1이되고 i가 짝수일 때 mode는 0이 되므로 answer += a; 를 두 번 쓸 필요가 없다.

	if (a != '1') {
       if (i % 2 == mode) {
          answer += a;
       }
    }

삼항연산자를 사용하면 mode 값도 짧은 코드로 바꿀 수 있다.

	if (a == '1') {
   		mode = mode == 0 ? 1 : 0;
       		continue;
   		}	
	}

answer가 빈 문자열일 경우 "EMPTY"를 반환하는 것도 삼항연산자를 사용하면 짧게 작성할 수 있다.

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++) {
            char a = code.charAt(i);
        
           if (a != '1') {
              if (i % 2 == mode) {
                answer += a;
              }
           }
           if (a == '1') {
               mode = mode == 0 ? 1 : 0;
                 continue;
               }
           }
            
        return answer.equals("") ? "EMPTY" : answer;
    }
}