[Python] 프로그래머스 코드 처리하기

문제 설명

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

 

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

프로그래머스

제한 사항

1 ≤ code의 길이 ≤ 100,000
code는 알파벳 소문자 또는 "1"로 이루어진 문자열입니다.

예제

code = "abc1abc1abc"
result = "acbac"

문제 풀이

def solution(code):
    ans = ''
    mode = 0
    
    for idx, val in enumerate(code):
        if val == "1":						# value가 1일 때
            mode = 0 if mode == 1 else 1	# mode 변경(삼항연산자)
        if mode == 0:						# mode가 0일 때
            if idx % 2 == 0 and val != "1": # 짝수 index의 value 추가
                ans += val
        else:								# mode가 1일 때
            if idx % 2 == 1 and val != "1":	# 홀수 index의 value 추가
                ans += val
    
    return ans if ans!="" else "EMPTY"

enumerate 함수는 iterable 자료형의 인덱스와 원소를 포함하는 enumerate 객체를 반환하는 함수입니다.

금번 문제처럼 인덱스와 원소 모두 사용할 때 활용하기 좋은 함수입니다.

L = enumerate([1,2,3,4,5])
for idx, value in L:
    print(idx, ":", value)
print(type(L))

output
0 : 1
1 : 2
2 : 3
3 : 4
4 : 5
<class 'enumerate'>

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.