logo

𝝅번째 알파카의 개발 낙서장

screen

[프로그래머스 / JAVA] Level 1 x만큼 간격이 있는 n개의 숫자 (12954)

posts

알고리즘

시리즈 톺아보기

프로그래머스

프로그래머스
count

x만큼 간격이 있는 n개의 숫자 🔗

랭크 사용 언어
Level 1 JAVA

🔗 x만큼 간격이 있는 n개의 숫자

문제 설명 🔗

함수 solution은 정수 x와 자연수 n을 입력 받아, x부터 시작해 x씩 증가하는 숫자를 n개 지니는 리스트를 리턴해야 합니다. 다음 제한 조건을 보고, 조건을 만족하는 함수, solution을 완성해주세요.

제한 사항 🔗

  • x는 -10000000 이상, 10000000 이하인 정수입니다.
  • n은 1000 이하인 자연수입니다.

입출력 예 🔗

x n answer
2 5 { 2, 4, 6, 8, 10 }
4 3 { 4, 8, 12 }
-4 2 { -4, -8 }

풀이 🔗

x만큼 간격을 가진 수를 n개 만큼 나열하여 배열로 반환한다. 다른건 문제가 없지만, 음수의 경우가 문제다.

음수의 경우 내림차순으로 정렬해야 한다.

ArrayList에 요소를 담은 후, 각 수의 절대값을 기준으로 오름차순으로 정렬하자. ArrayList를 스트림화하여 sorted() 메서드를 사용하되, 정렬 알고리즘을 오버라이딩한다. (o1, o2) -> (int) (Math.abs(o1) - Math.abs(o2))로 오버라이딩하면 된다. Math.abs() 메서드를 통해 각 수의 절대값을 추출할 수 있다.

코드 🔗

JAVA

0import java.util.ArrayList;
1
2/**
3 * x만큼 간격이 있는 n개의 숫자 클래스
4 *
5 * @author RWB
6 * @since 2021.12.13 Mon 22:21:01
7 */
8class Solution
9{
10 /**
11 * 해답 반환 메서드
12 *
13 * @param x: [int] 간격
14 * @param n: [int] 갯수
15 *
16 * @return [long[]] 해답
17 */
18 public long[] solution(int x, int n)
19 {
20 ArrayList<Long> list = new ArrayList<>();
21
22 while (n != 0)
23 {
24 list.add((long) x * n);
25
26 n--;
27 }
28
29 return list.stream().sorted((o1, o2) -> (int) (Math.abs(o1) - Math.abs(o2))).mapToLong(Long::longValue).toArray();
30 }
31}