[Kotlin/프로그래머스] 코딩테스트 연습 > 탐욕법(Greedy) > 체육복

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
class Solution {
    fun solution(n: Int, lost: IntArray, reserve: IntArray): Int {
        val array = Array<Int>(n) { 1 }
        lost.forEach {
            array[it - 1]--
        }
        reserve.forEach {
            array[it - 1]++
        }
        array.forEachIndexed { index, i ->
            if (i > 1 && array.getOrNull(index - 1) == 0) {
                array[index - 1] = 1
                array[index]--
            } else if (i > 1 && array.getOrNull(index + 1) == 0) {
                array[index + 1] = 1
                array[index]--
            }
        }
        return array.count { it != 0 }
    }
}