문자열 code가 주어집니다.
code를 앞에서부터 읽으면서 만약 문자가 "1"이면 mode를 바꿉니다. mode에 따라 code를 읽어가면서 문자열 ret을 만들어냅니다.
mode는 0과 1이 있으며, idx를 0 부터 code의 길이 - 1 까지 1씩 키워나가면서 code[idx]의 값에 따라 다음과 같이 행동합니다.
문자열 code를 통해 만들어진 문자열 ret를 return 하는 solution 함수를 완성해 주세요.
단, 시작할 때 mode는 0이며, return 하려는 ret가 만약 빈 문자열이라면 대신 "EMPTY"를 return 합니다.
code | result |
---|---|
"abc1abc1abc" | "acbac" |
따라서 "acbac"를 return 합니다.
def solution(code):
result = ''
mode = 0
for i in range(len(code)):
if mode == 0 :
if code[i] != '1' and i % 2 == 0:
result += code[i]
elif code[i] == '1':
mode = 1
else :
if code[i] != '1' and i % 2 == 1 :
result += code[i]
elif code[i] == '1':
mode = 0
if result == '':
return 'EMPTY'
return result
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'