개발자가 되

프로그래머스 Lv.0

Lv.0 코드 처리하기

dltjdud 2025. 1. 24. 22:44

문제설명

문자열 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가 변합니다.

icode[i]moderet

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'){
                    if(i%2==0){
                        answer+=code.charAt(i);
                    }
                }else if(code.charAt(i)=='1'){
                    mode=1;
                } 
            }else{
                if(code.charAt(i)!='1'){
                    if(i%2==1){
                        answer+=code.charAt(i);
                    }
                }else if(code.charAt(i)=='1'){
                    mode=0;
                }
            }

        }

        if(answer.equals("")){
            answer = "EMPTY";
        }
        return answer;
    }
}

* 좋은 풀이

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();
    }
}

 

Python

def solution(code):
    answer = ''

    mode = 0
    for i in range(len(code)):
        if code[i] == '1':
            mode ^= 1
        else:
            if i % 2 == mode:
                answer += code[i]

    return answer if answer else 'EMPTY'

*한줄 코딩

def solution(code):
    return "".join(code.split("1"))[::2] or "EMPTY"

 

JS

function solution(code) {
    let answer = '';
    let mode = 0;

    for (let i = 0; i < code.length; i += 1) {
      if (Number(code[i]) === 1) {
        mode = mode === 1 ? 0 : 1;
      }
      if (Number(code[i]) !== 1 && i % 2 === mode) {
        answer += code[i];
      }
    }
    return answer.length > 0 ? answer : 'EMPTY';
}

*한줄 코딩

function solution(code) {
    var answer = code.replaceAll("1","").split("").filter((val, idx)=> idx%2 === 0).join("");
    return answer.length > 0 ? answer : "EMPTY";
}

'프로그래머스 Lv.0' 카테고리의 다른 글

Lv.0 주사위 게임 2  (0) 2025.01.25
Lv.0 원소들의 곱과 합  (0) 2025.01.25
Lv.0 등차수열의 특정한 항만 더하기  (0) 2025.01.24
Lv.0 flag에 따라 다른 값 반환하기  (0) 2025.01.23
Lv.0 조건 문자열  (0) 2025.01.23