[프로그래머스] Lv.0 - 코드 처리하기.py

문제

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

입출력 예

입출력 설명

• 입출력 예 1
  • code의 각 인덱스 i에 따라 다음과 같이 mode와 ret가 변합니다.

• 따라서 "acbac"를 return 합니다. 
• 입출력 예 2
  • 10 이상 20 이하이면서 0과 5로만 이루어진 정수는 없습니다. 따라서 [-1]을 return 합니다.

코드

def solution(l, r):
    answer = []


    num_list = []
    check_num = ['0', '5']
    for num in range(l, r+1):
        if all(one_num in check_num for one_num in str(num)):
            answer.append(num)
    if len(answer) == 0:
        answer.append(-1)

    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges