[Python] 프로그래머스 배열 만들기 2

문제 설명

정수 l과 r이 주어졌을 때, l 이상 r이하의 정수 중에서 

숫자 "0"과 "5"로만 이루어진 모든 정수를 오름차순으로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.

만약 그러한 정수가 없다면, -1이 담긴 배열을 return 합니다.

 

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

프로그래머스

제한 사항

1 ≤ l ≤ r ≤ 1,000,000

예제

# 1
l      = 5
r      = 555
result = [5, 50, 55, 500, 505, 550, 555]

# 2
l      = 10
r      = 20
result = [-1]

문제 풀이

# 1
def solution(l, r):
    answer = []
    for i in range(l, r+1):
        num = set(str(i))
        if num == {'0','5'} or num == {'0'} or num == {'5'}:
            answer.append(i)
    return answer if len(answer) > 0 else [-1]
    
# 2
def solution(l, r):
    answer = []
    for i in range(l, r+1):
        if not set(str(i))-{'0','5'}:
            answer.append(i)
    return answer if len(answer) > 0 else [-1]

---

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