[Kotlin] 제일 작은 수 더하기 (프로그래머스 알고리즘)

 

 

 

문제

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

 

 

---

 

접근

 

• 1. arr에서 제일 작은 수를 찾고 (null일 수도 있으니 .min() 보다는 .minOrNull() 사용) arr에서 제거
• 2. arr가 비지 않았을 경우 arr 리턴, arr가 비었을 경우 [-1] 리턴

class Solution {
    fun solution(arr: IntArray): IntArray {
        var answer:IntArray = arr - arr.minOrNull()
        if(answer.size == 1) return intArrayOf(-1) else return answer
    }
}

• 위 수식처럼 구상하였는데 오류가 났다.
• ' - '연산자를 사용하여 배열에서 다른 배열을 빼는 것은 허용되지 않기 때문이다.
• 대신에 filter()를 이용하여 배열의 요소를 수정하는 방식으로 작성해야 한다.

 

---

 

성공 코드

 

class Solution {
    fun solution(arr: IntArray): IntArray {
        return if (arr.size == 1) intArrayOf(-1) 
        else arr.filter { it != arr.minOrNull() }.toIntArray()
    }
}

 

---

 

설명

 

1. if (arr.size == 1) intArrayOf(-1) else ...: 
   배열의 길이가 1인 경우를 처리한다. 만약 배열의 길이가 1이면, 배열에는 최솟값만 존재하므로 배열이 [-1]이 되어야 한다. 따라서 이 부분은 삼항 연산자를 사용하여 배열의 길이가 1인 경우에는 -1을 포함한 배열을 반환한다.
2. arr.filter { it != arr.minOrNull() }
   배열에서 최솟값을 제외한 요소들로 이루어진 새로운 리스트를 생성한다. filter() 함수를 사용하여 주어진 조건을 만족하는 요소들만을 선택하여 새로운 리스트를 생성하는데, 여기서는 배열의 최솟값과 같지 않은 요소들을 선택한다.
3. .toIntArray()
   새로운 리스트를 정수 배열로 변환한다. Kotlin에서 filter() 함수는 기본적으로 리스트를 반환하므로, 배열로 변환하기 위해 toIntArray() 함수를 호출한다.

 

---

 

다른 방법

 

class Solution {
    fun solution(arr: IntArray): IntArray {
        var min = Int.MAX_VALUE
        for(i in arr) {
            if(i < min) min = i
        }

        var array = ArrayList<Int>()

        for(i in arr) {
            if(i != min) array.add(i)
        }

        return if(arr.size == array.size || array.isEmpty())  {
            IntArray(1) {-1}
        } else array.toIntArray()
    }
}

• minOrNull()과 filter()를 사용하지 않을 때 일일히 계산하는 방법이다.