[프로그래머스] 파이썬 배열 만들기 2

ybk·2023년 9월 21일
0

프로그래머스

목록 보기
29/236

Lv.0 배열 만들기 2

문제 설명

정수 l과 r이 주어졌을 때, l 이상 r이하의 정수 중에서 숫자 "0"과 "5"로만 이루어진 모든 정수를 오름차순으로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.
만약 그러한 정수가 없다면, -1이 담긴 배열을 return 합니다.

입출력

코드 풀이

def solution(l, r):
    answer = []
    for num in range(l, r+1):
        if not set(str(num)) - {'0','5'}:
            answer.append(num)
    return answer if answer else [-1]
  • set(str(num))은 숫자를 문자열로 변환한 후 해당 문자열의 문자들을 집합으로 만들고 이 집합에서 '0'과 '5'를 제외한 문자들을 뺀 집합({'0', '5'}와의 차집합)을 계산한다.

출처: 프로그래머스 코딩테스트 연습

profile
개발자 준비생~

0개의 댓글