[프로그래머스] 코드 처리하기 - JAVA

프로그래머스 _ 코딩 기초 트레이닝 DAY 5 - (1)

출처 - https://school.programmers.co.kr/learn/courses/30/lessons/181932

프로그래머스

 

문제 | 문자열 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 | "abc1abc1abc"

출력 #1 | "acbac"

---

>> 풀이 << 

빈 문자열을 생성하여 계속 문자를 추가하면 앞서 공부했듯이 메모리 낭비가 심할 수 있기 때문에 StringBuilder를 생성해 코드를 작성했다. if 문을 이용하여 mode가 0일 때, 1일 때를 나누고 그 안에서 문제 조건을 활용하였다.

 

이 문제를 풀면서 앞서 공부한 것들을 총체적으로 활용했다.

• String에서 문자 한개씩 가져오기 -> charAt()
• StringBuilder 생성, 스트링빌더에 문자 추가 -> append() , 꼭 마지막에는 스트링빌더에 있는 것을 문자열로 바꾸기 -> toString()
• 문자열 비교 -> equals() 

 

문제를 풀 때 마지막 조건인 "빈 문자열이라면 대산 EMPTY를 return" 을 못봐서 오래 걸렸다,, 문제는 꼭 끝까지 읽기!

또 리턴할 문자열이 null인지 확인할 때 '== null' 이 아니라 '.equals("")' 라고 써야한다는 거 기억하기!

---

최종 코드는 이렇게 작성했다.

class Solution {
    public String solution(String code) {
        
        int mode = 0;
        StringBuilder sb = new StringBuilder();
        
        for (int i = 0 ; i < code.length() ; i++) {
            if (mode == 0) {
                if (code.charAt(i) == '1') mode = 1;
                else {
                    if (i % 2 == 0) sb.append(code.charAt(i));
                }
            }
            else if (mode == 1) {
                if (code.charAt(i) == '1') mode = 0;
                else {
                    if (i % 2 == 1) sb.append(code.charAt(i));
                }
            }
        }
        
        if (sb.toString().equals("")) return "EMPTY";
        return sb.toString();
    }
}