[프로그래머스] 배열 만들기 2 Java

문제

정수 l과 r이 주어졌을 때, l 이상 r이하의 정수 중에서 숫자 "0"과 "5"로만 이루어진 모든 정수를 오름차순으로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.
만약 그러한 정수가 없다면, -1이 담긴 배열을 return 합니다.

 

제한사항
1 ≤ l ≤ r ≤ 1,000,000

 

입출력 예 #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 int[] solution(int l, int r) {
        ArrayList<Integer> arrayList = new ArrayList<>();
        
        for (int i = l; i <= r; i++) {
            String str = i + "";
            int count = 0;
            for (int j = 0; j < str.length(); j ++) {
                if (str.charAt(j) == 48 || str.charAt(j) == 53) {
                    count++;
                }
            }
            
            if (count == str.length()) {
                arrayList.add(i);
            }
        }
        int[] answer = new int[arrayList.size()];
        
        if (answer.length == 0) {
            return new int[]{-1};
        }
        for (int i = 0; i < arrayList.size(); i++) {
            answer[i] = arrayList.get(i);
        }
        return answer;
    }
}