Your Name

I am an amazing person.

1 분 소요

문제 설명

프로그래머스 181921번 배열 만들기 2

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

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

제한 사항

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

입출력 예시

예시 1

입출력 1

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 합니다.

코드 구현

Python

def solution(l, r):
    answer = []
    
    for i in range(l, r + 1):
        for idx, s in enumerate(str(i)):
            if s != "0" and s != "5":
                break
            if idx == len(str(i)) - 1:
                answer.append(i)
    
    return [-1] if len(answer) == 0 else answer

Java

import java.util.ArrayList;

class Solution {
    public int[] solution(int l, int r) {
        ArrayList<Integer> answer = new ArrayList<>();

        for (int i = l; i <= r; i += 1) {
            char[] number = String.valueOf(i).toCharArray();
            for (int index = 0; index < number.length; index += 1) {
                char c = number[index];
                if (c != '0' && c != '5') {
                    break;
                }

                if (index == number.length - 1) {
                    answer.add(i);
                }
            }
        }

        return answer.isEmpty() ? new int[]{-1} : answer.stream().mapToInt(i -> i).toArray();
    }
}

참고