[Kotlin] x만큼 간격이 있는 n개의 숫자 (프로그래머스 알고리즘)

 

 

https://school.programmers.co.kr/learn/courses/30/lessons/12954

 

 

 

[오류 코드] 
테스트 14개 중 13, 14번 테스트 통과 못한 코드



class Solution {
    fun solution(x: Int, n: Int): IntArray {
        
        var list = IntArray(n)
        for (i in 0..n-1){
            list[i] = x * (i+1)
        }
        
        
        return list
    }
}

[통과 코드]
LongArray로 해결



class Solution {
    fun solution(x: Int, n: Int): LongArray {
        
        var list = LongArray(n)
        for (i in 0..n-1){
            list[i] = x.toLong() * (i+1)
        }
                     
        return list
    }
}

 

설명

1. 먼저 n의 크기만큼 배열을 생성한다.
2. 배열에 값을 넣어주기 위해 for문을 이용하여 인덱스 0부터 n-1까지 반복한다.
3. 각 인덱스에 해당하는 값을 알맞은 수식을 작성하여 넣어준다.
4. 배열을 반환한다.

 

오류가 난 이유

• 오버플로우를 생각해야 했다.
• IntArray로 작성하기에는 테스트 값이 해당 범위를 넘어갔기에 Long으로 설정해야 했다.
• 형변환을 (x * i).toLong() 식으로 처리하면 int형으로 계산되고 그 다음에 long으로 변경된 값이 들어간다.
• 따라서 x.toLong() * i  식으로 해야 한다.

 

 

 

class Solution {
    fun solution(x: Int, n: Int): LongArray = LongArray(n) { x.toLong() * (it + 1) }
}

 

• 이렇게 간단히 코드를 작성하는 방법도 있다.
• n 크기의 LongArray를 선언하고, 각 값에 알맞은 값을 넣을 수식을 입력한다.
• ❓ it은 람다식 내부에서 사용되는 암시적 변수이다.