[코딩테스트 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"로 이루어진 문자열입니다.

입출력 예

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 합니다.

---

내 풀이

Java

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

C

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

// 파라미터로 주어지는 문자열은 const로 주어집니다. 변경하려면 문자열을 복사해서 사용하세요.
char* solution(const char* code) {
    // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
    char* answer = (char*)malloc(sizeof(char) * strlen(code));
    int i, k;
    int mode = 0;
    for(i = 0, k = 0; i < strlen(code); i++){
        if(mode == 0){
            if(code[i] != '1' && i % 2 == 0){
                answer[k++] = code[i];
            }else if(code[i] == '1'){
                mode = 1;    
            }
        }else{
            if(code[i] != '1' && i % 2 != 0){
                answer[k++] = code[i];
            }else if(code[i] == '1'){
                mode = 0;    
            }
        }
    }
    answer[k] = '\0';
    
    if(strlen(answer) == 0) {
        return "EMPTY";        
    }
    
    return answer;
}

Python

def solution(code):
    answer = ''
    mode = 0
    i = 0
    for c in code:
        if mode == 0:
            if c != '1' and i % 2 == 0:
                answer += c
            elif c == '1':
                mode = 1
        else:
            if c != '1' and i % 2 != 0:
                answer += c
            elif c == '1':
                mode = 0
        i += 1
        
    if answer == '':
        answer = 'EMPTY'
    return answer

---

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

프로그래머스