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

 

import java.util.*;

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

        while (num <= r) {
            String tmp = String.valueOf(num);

            // 0 또는 5로만 이루어져 있는지 확인
            boolean valid = true;
            for (int i = 0; i < tmp.length(); i++) {
                if (tmp.charAt(i) != '0' && tmp.charAt(i) != '5') {
                    valid = false;
                    break;
                }
            }

            // 0 또는 5로만 이루어져 있으면 스택에 추가
            if (valid) {
                st.push(num);
            }

            num += 1; // 한 번에 5씩 증가하는 대신 1씩 증가하도록 수정
        }

        if (st.isEmpty()) { // 스택이 비어있다면 
            answer = new int[]{-1}; // answer 초기화 후 -1 저장
        } else {
            answer = new int[st.size()]; // 스택이 비어있지 않다면 st 사이즈만큼 크기할당
            for (int i = 0; i < answer.length; i++) {
                answer[i] = st.pop(); // 해당 값들을 역순으로 저장
            }
            Arrays.sort(answer); // 역순으로 저장된 값들을 오름차순 정렬
        }
        return answer;
    }
}

 

---

0또는 5로 끝나는 수인지 확인하라고해서

처음엔 

num을 5씩 증가하게 했었다.

num += 5 -> num += 1로 바꾸니 통과가 되었는데

이유는 좀더 분석을 해봐야겠다