[Programmers] Lv1.체육복(kotlin)

문제

https://programmers.co.kr/learn/courses/30/lessons/42862?language=kotlin

예시

https://programmers.co.kr/learn/courses/30/lessons/42862?language=kotlin

 

풀이

1. 체육복을 도난당하는 학생(lost)와 여벌의 옷이 있는 학생(reserve) 에서 같은 번호는 제거합니다.(마지막 제한사항)

2. 같은 번호가 제거된 reallost 리스트와 realreserve 리스트를 번호가 큰 크기 순으로 정렬하고 reallost 리스트는 여벌의 옷이 있을 경우 번호 삭제가 가능하도록 mutableList로 변경합니다.

3. realreserve 리스트를 탐색하여 도난당한 학생에게 빌려줄 수 있는지 여부를 파악한 후 빌려줄 수 있으면 reallost에서 학생의 번호를 제거합니다.

 

class Solution {
    fun solution(n: Int, lost: IntArray, reserve: IntArray): Int {
        var answer = n
        val reallost = lost.filter{ n -> reserve.count{ it == n} == 0 }.sortedDescending().toMutableList()
        val realreserve = reserve.filter{ n -> lost.count{ it == n} == 0}.sortedDescending()

        realreserve.forEach{ num ->
            if(num == n){
                if(reallost.count{ num-1 == it} > 0){
                    reallost.remove(num-1)
                }
            }else if(num == 1){
                if(reallost.count{ num+1 == it} > 0){
                    reallost.remove(num+1)
                }
            }else{
                if(reallost.count{ num+1 == it } > 0){
                    reallost.remove(num+1)
                }else{
                    if(reallost.count{ num-1 == it} > 0){
                        reallost.remove(num-1)
                    }
                }
            }
        }
        
        
        return answer-reallost.size
    }
}

 

참고

• 함수형 프로그래밍 언어인 kotlin의 특성에 맞게 함수를 많이 익히자 😽
• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/count.html

count - Kotlin Programming Language

• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/sorted-descending.html

sortedDescending - Kotlin Programming Language