main

main.kt

fun getTimes(times: Array<Int>, directions: Array<Int>): Array<Int> {
    val result: Array<Int> = Array(times.size) { -1 }

    val endTime: Int = times.max() ?: 0
    if (endTime == 0) return result
    val inQueue: Queue<Int> = LinkedList()
    val outQueue: Queue<Int> = LinkedList()

    var currentIndex = 0
    var turnstile = 0

    for (time in 0 until endTime + 1) {
        while (times[currentIndex] == time) {
            if (directions[currentIndex] == 0) {
                inQueue.add(currentIndex)
            }
            if (directions[currentIndex] == 1) {
                outQueue.add(currentIndex)
            }
            currentIndex++
            if (currentIndex == times.size) break
        }

        val noPersonOnTurn = inQueue.size + outQueue.size == 0
        if (noPersonOnTurn) turnstile = 0

        if (turnstile == 1 || outQueue.size == 0) {
            if (inQueue.size > 0) {
                val next = inQueue.poll()
                result[next] = time
                turnstile = 1
            }
        } else {
            if (outQueue.size > 0) {
                val next = outQueue.poll()
                result[next] = time
                turnstile = 2
            }
        }
    }
    return result
}