[Algorithm] 프로그래머스 - 배열의 원소 삭제하기 (Kotlin)

정수 배열 arr, delete_list가 있을 때 arr의 원소 중 delete_list의 원소를 모두 삭제하고 남은 원소들은 기존의 arr에 있던 순서를 유지한 배열을 리턴하는 함수를 완성하라

 

 

간단하게 풀면 아래와 같이 풀 수 있다.

 

class Solution {
    fun solution(arr: IntArray, deleteList: IntArray) = arr.toList() - deleteList.toSet()
}

 

컬렉션 함수를 쓰는 방법도 있다.

 

class Solution {
    fun solution(arr: IntArray, delete_list: IntArray): IntArray =
        arr.map { arr.filter { !delete_list.contains(it) } }
            .flatten()
            .distinct()
            .toIntArray()
}