코딩테스트

코테 Lv0 - 배열 만들기 2

Beekei 2024. 11. 25. 21:47
반응형

문제 설명

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

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

제한사항

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

입출력 예

입출력 예 #1

Input : 5(l), 555(r) / Output : [5, 50, 55, 500, 505, 550, 555]

더보기

5 이상 555 이하의 0과 5로만 이루어진 정수는 작은 수부터 5, 50, 55, 500, 505, 550, 555가 있습니다. 

따라서 [5, 50, 55, 500, 505, 550, 555]를 return 합니다.

입출력 예 #2

Input : 10(l), 20(r)  / Output : [-1]

더보기

10 이상 20 이하이면서 0과 5로만 이루어진 정수는 없습니다. 따라서 [-1]을 return 합니다.

제출코드

import java.util.ArrayList;
import java.util.List;

class Solution {
    public int[] solution(int l, int r) {
        int min = Integer.parseInt("1" + ("0".repeat(String.valueOf(l).length() - 1)), 2);
        int max = Integer.parseInt("1".repeat(String.valueOf(r).length()), 2);
        int[] answer = IntStream.rangeClosed(min, max)
            .map(num -> Integer.parseInt(Integer.toBinaryString(num)) * 5)
            .filter(num -> l <= num && r >= num)
            .toArray();
        return answer.length == 0 ? new int[] { -1 } : answer;
    }
}
  1. 주어진 최소, 최댓값의 length를 활용해 최소, 최대 이진수로 변환 후 십진수로 변환합니다.
  2. 최소부터 최대까지 증가하며 2진수로 변환 후 5를 곱하여 반환합니다.
  3. 주어진 최소, 최대값에 포함되는 숫자만 필터링하여 배열로 변환 후 반환합니다.
  4. 이렇게 만들어진 배열에 길이가 0이라면 -1을 가진 배열, 길이가 0보다 크다면 배열을 반환합니다.

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181921

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

반응형