본문 바로가기

앱 개발/Algorithm

minOrNull(), filter (제일 작은 수 제거하기)

< 문제 29번 >

 정수를 저장한 배열 arr에서 가장 작은 수를 제거한 배열을 리턴하는 solution 함수를 완성해주세요.

리턴하려는 배열이 빈 배열인 경우에는 배열에 -1을 채워 리턴하세요. 예를 들어 arr이 [4,3,2,1]인 경우는 [4,3,2]를 리턴하고, [10]이면 [-1]을 리턴합니다.

 

 

< solution >

 

class Solution {
    fun solution(arr: IntArray): IntArray {
        if (arr.size <= 1) {
            return intArrayOf(-1)
        }

        // arr 배열에서 최소값 찾기. 배열이 비었을 경우에는 null을 반환하지만 문제 조건에 따라 [-1]을 반환      

        val min = arr.minOrNull()

 

        // 최소값을 제외한 요소들을 필터링하여 정수 배열로 변환하여 반환
        return arr.filter { it != min }.toIntArray()
    }
}

 

 

(참고) https://lionbae.tistory.com/entry/indexOf-filter

728x90