[Lv. 1] x만큼 간격이 있는 n개의 숫자

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/12954

프로그래머스

난이도 : Level.1

언어 : Kotlin

 

문제 설명 :

함수 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]

풀이 : 

import kotlin.math.abs

class Solution {
    fun solution(x: Int, n: Int): LongArray {
        var temp = x.toLong()

        return try {
            (temp .. temp * n).step(temp).map { it }.toLongArray()
        } catch (_: Exception) {
            temp = abs(x).toLong()

            if (x == 0) {
                LongArray(n) { 0 }
            } else {
                (temp .. temp * n).step(temp).map { -it }.toLongArray()
            }
        }
    }
}