[Kotlin/프로그래머스] n 번째 원소까지

문제

정수 리스트 num_list와 정수 n이 주어질 때, num_list의 첫 번째 원소부터 n 번째 원소까지의 모든 원소를 담은 리스트를 return하도록 solution 함수를 완성해주세요.

 

 풀이 

class Solution {
    fun solution(num_list: IntArray, n: Int): IntArray {
        var answer: MutableList<Int> = mutableListOf<Int>()
        
        // for (i in 0..n)
        for(i in 0 until n){
            answer.add(num_list[i])
        }
        
        return answer.toIntArray()
    }
}

 

 다른 사람 풀이 

class Solution {
    fun solution(num_list: IntArray, n: Int): IntArray {
        var answer: IntArray = intArrayOf()
        for (i in 0 until n) {
            answer += num_list[i]
        }
        return answer
    }
}

 

주석에 달아 놓은 것 처럼 for문을 돌리면 해당 번째 값 까지 포함되니까 주의!

그리고 배열은 0번째부터 시작하니까 시작 숫자를 1이 아닌 0으로 해야한다.

상당히 기초적인 주의사항이지만.. 또 틀려버림!

 

아, 그리고 마지막에 return 값이 IntArray 가 되어야 하니까 변환해주는거 잊지 않기~

그냥 처음부터 intArray  쓰고 더하는 방식으로 할껄.. 왜 mutableList 사용했는지 나도 의문 (??)