[프로그래머스] 다리를 지나는 트럭 (Kotlin)

문제 : programmers.co.kr/learn/courses/30/lessons/42583

코딩테스트 연습 - 다리를 지나는 트럭

 

 

import java.util.*

class Solution {
fun solution(bridge_length: Int, weight: Int, truck_weights: IntArray): Int {
    var answer = 0

    // 다리를 건너는 트럭
    val bridgeQueue : Queue<Int> = LinkedList<Int>()
    // 대기 트럭
    val waitQueue : Queue<Int> = LinkedList<Int>()

    // 다리 길이만큼 임의의 숫자 0을 채워줌
    for(i in 0 until bridge_length) bridgeQueue.add(0)
    for(i in truck_weights) waitQueue.add(i)

    // 모두 통과할 때 까지
    while(bridgeQueue.isNotEmpty()){
        answer++
        bridgeQueue.poll()
        if(waitQueue.isNotEmpty()) {
                // 견딜수 있는 무게를 초과하지 않는다면 큐에 삽입
                if (bridgeQueue.sum() <= weight - waitQueue.peek()) {
                    bridgeQueue.add(waitQueue.poll())
                } else {
                    bridgeQueue.add(0)
                }
            }
        }
        return answer
}
}