[Programmers] 코드 처리하기

문제 설명

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

 

---

문제 풀이

접근 전략

주어진 조건에 따라 문자열을 순회하면서 mode를 변경하고, 그에 맞게 적절하게 문자를 추가하는 방식으로 코드를 적어내려가면 됨.

 

코드

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

결과 및 해석

위의 코드는 주어진 문자열을 순회하며 각 문자에 대해 mode에 따라 동작을 결정함. 만약 문자(char)가 "1"이라면, mode를 변경하고, 그렇지 않으면 mode와 index의 짝수/홀수 여부에 따라 어떻게 문자를 추가할지 작성함. 마지막으로 만들어진 문자열이 비어있지 않다면 해당 문자열을, 비어있다면 "EMPTY"를 반환하여 끝냄. 시간 복잡도는 O(n)으로, 주어진 문자열의 길이에 비례함.

 

chatGPT 형님은 mode의 변경을 mode = 1 - mode로 더 간단히 표현함. 이렇게 하면 if-elif 문의 사용을 줄일 수 있으며 코드의 가독성이 높아짐.

 

def solution(code):
    answer = ''
    mode = 0
    
    for i, char in enumerate(code):
        if char == "1":
            mode = 1 - mode
        elif mode == 0 and i % 2 == 0:
            answer += char
        elif mode == 1 and i % 2 == 1:
            answer += char
            
    return answer if answer else "EMPTY"