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

문제 설명

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

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

---

---

입출력 예 설명

입출력 예 #1

• 5 이상 555 이하의 0과 5로만 이루어진 정수는 작은 수부터 5, 50, 55, 500, 505, 550, 555가 있습니다. 따라서 [5, 50, 55, 500, 505, 550, 555]를 return 합니다.

입출력 예 #2

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

---

풀이)

def solution(l, r):
    answer = []
    for i in range(l,r+1):
        if set(str(i)) <= {'0', '5'}:
                answer.append(i)
        if len(answer) == 0:
            return[-1]
    return answer

# {'0', '5'} : 문자열 0과 5로 이루어진 요소를 가지는 집합

 

# if set(str(num)) <= {'0', '5'} : 숫자 num을 문자열로 변환한 후, 이 문자열이 '0'과 '5'로만 이루어져 있는지를 검사하는 조건