문자열 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 합니다.
code
의 길이 ≤ 100,000code
는 알파벳 소문자 또는 "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" |
def solution(code):
answer = ''
mode = 0
for idx, s in enumerate(code):
# mode가 1일 경우
if mode == 1:
# 현재 문자가 '1'인 경우
if s == '1':
mode = 0
# 현재 문자가 '1'이 아닌 경우
else:
if idx % 2 == 1:
answer += s
# mode가 0일 경우
else:
# 현재 문자가 '1'인 경우
if s == '1':
mode = 1
# 현재 문자가 '1'이 아닌경우
else:
if idx % 2 == 0:
answer += s
if answer == '':
return 'EMPTY'
return answer
https://school.programmers.co.kr/learn/courses/30/lessons/181932?language=python3