[프로그래머스] 체육복(42862)(Kotlin)

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

코딩테스트 연습 - 체육복 | 프로그래머스

 

<첫번째> 2020/06/08

class Solution {
    fun solution(n: Int, lost: IntArray, reserve: IntArray): Int {

        val list = MutableList(n) { 1 }

        lost.forEach { list[it - 1]-- }
        reserve.forEach { list[it - 1]++ }

        list.forEachIndexed { index, value ->
            if (value == 0) {
                if (index - 1 > 0 && list[index - 1] > 1) {
                    list[index - 1]--
                    list[index]++
                } else if (index + 1 < n && list[index + 1] > 1) {
                    list[index + 1]--
                    list[index]++
                }
            }
        }

        return list.count { it > 0 }
    }

}