[Programmers | Level 0] 배열 만들기 2

IBLOG·2023년 7월 17일
0

programmers

목록 보기
26/26

programmers

문제 설명

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

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

제한사항

1 ≤ l ≤ r ≤ 1,000,000

입출력 예

lrresult
5555[5, 50, 55, 500, 505, 550, 555]
1020[-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 합니다.



Solution Code

# Solution 1

def solution(l, r):
    result = []

    for i in range(l, r+1):
        if i % 5 != 0:
            continue

        num = str(i) 
        is_zero_or_five = True

        for j in num:
            if j != "5" and j != "0":            
                is_zero_or_five = False
        if is_zero_or_five:
            result.append(i)

    if not result:
        return [-1]
    return result

# Solution 2

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]
profile
✢ 천천히 하나씩 기록해보자!

0개의 댓글