Your Name

I am an amazing person.

1 분 소요

문제 설명

프로그래머스 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 ≤ code의 길이 ≤ 100,000
    • code는 알파벳 소문자 또는 “1”로 이루어진 문자열입니다.

입출력 예시

예시 1

입출력 1

code result
“abc1abc1abc” “acbac”

입출력 예 설명 1

  • code의 각 인덱스 i에 따라 다음과 같이 mode와 ret가 변합니다.
i code[i] mode ret
0 “a” 0 “a”
1 “b” 0 “a”
2 “c” 0 “ac”
3 “1” 1 “ac”
4 “a” 1 “ac”
5 “b” 1 “acb”
6 “c” 1 “acb”
7 “1” 0 “acb”
8 “a” 0 “acba”
9 “b” 0 “acba”
10 “c” 0 “acbac”

따라서 “acbac”를 return 합니다.

코드 구현

Python

def solution(code):
    answer = ''
    mode = "0"
    
    for i in range(len(code)):
        if code[i] == "1":
            if mode == "0":
                mode = "1"
            else:
                mode = "0"
        else:
            if i % 2 == 0 and mode == "0":
                answer += code[i]
            elif i % 2 == 1 and mode == "1":
                answer += code[i]

    answer = "EMPTY" if len(answer) == 0 else answer
    
    return answer

Java

class Solution {
    public String solution(String code) {
        String answer = "";

        int mode = 0;

        for (int i = 0; i < code.length(); i+=1) {

            Character c = code.charAt(i);

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

        return answer;
    }
}

참고