[프로그래머스] 코딩 기초 트레이닝 > 배열 만들기2

문제 보기

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

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

---

제한사항

• 1 ≤ l ≤ r ≤ 1,000,000

---

입출력 예

l                    r                         result

5	555	[5, 50, 55, 500, 505, 550, 555]
10	20	[-1]

---

입출력 예 설명

입출력 예 #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 not answer:
        return [-1]
    return answer

set(str(i))는 정수 i를 문자열로 변환한 후, 그 문자열을 집합(set)으로 변환하는 과정입니다. 예를 들어, i가 105라면, str(i)는 "105"가 되고, set(str(i))는 {'1', '0', '5'} 집합이 됩니다.

그리고 <= 연산자는 부분집합(subset) 관계를 나타내며, 왼쪽의 집합이 오른쪽의 집합의 부분집합인 경우에 참(True)을 반환합니다. 따라서 set(str(i)) <= {'0', '5'}는 정수 i의 각 숫자가 '0' 또는 '5'로만 이루어져 있는지를 검사합니다.

예를 들어, i가 105이면, set(str(i))는 {'1', '0', '5'}이고, 이는 {'0', '5'}의 부분집합입니다. 따라서 조건식이 참이 되어 해당 정수를 결과 리스트에 추가합니다.