[프로그래머스] 배열 만들기 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 합니다.

제출 답안

• 단순하게 중복 순열을 사용하여 풀이하였다.

from itertools import product

def solution(l, r):
    answer = set()
    combi = [product(["0", "5"], repeat=i) for i in range(1, len(str(r)) + 1)]
    for com in combi:
        for c in com:
            num = int("".join(c))
            if l <= num <= r:
                answer.add(num)
    answer = sorted(list(answer))
    return answer if answer else [-1]

차집합 사용

• 차집합을 사용하면 0과 5중에 조건에 해당되는 것들이 없어지면 빈 집합이 나온다.
• 그것을 기준으로 값을 구하면 된다.

def solution(l, r):
    answer = []
    for num in range(l, r + 1):
        if not set(str(num)) - set(['0', '5']):
            answer.append(num)
    return answer if answer else [-1]