Programmers: 다리를 지나는 트럭

54:45

푸는 데 정말 오래걸렸다. 사실 이 문제는 예전에도 몇 번 시도했다가 실패했는데 이번에는 성공!

1. 일단, 처음에는 대기 트럭 목록이 비어있고, 큐(다리)가 빌 때까지 반복하게 했는데 이유는 몰라도 동작하지 않아서 도착한 트럭의 수가 입력으로 들어온 대기 트럭 수와 같을 때까지 반복하기로 했다.
2. 다리를 건너고 있는 트럭이 있는지 확인한다. 어차피 1초에 한 대씩밖에 못 움직이니 가장 첫 트럭만 확인하면 된다.
3. 다리에 있는 트럭의 무게와 대기 트럭의 맨 앞에 있는 트럭의 무게를 합했을 때, 초과하지 않으면 queue에 넣어준다. 이 때, 현재 시간을 같이 넣어서 2에서 시간을 확인할 때 쓴다.
   
   

문제

https://programmers.co.kr/learn/courses/30/lessons/42583
https://www.acmicpc.net/problem/13335

내가 작성한 코드

Swift

import Foundation  

func solution(_ bridge_length:Int, _ weight:Int, _ truck_weights:[Int]) -> Int {
    var waits = truck_weights // 대기 트럭
    var queue = [[Int]]() // 다리를 건너는 트럭 
    var finished = [[Int]]() // 다리를 지난 트럭
    var time = 0 // 경과 시간
    while finished.count != truck_weights.count { // 도착 트럭 목록 수가 입력으로 들어온 대기 트럭 수와 같을 때까지 반복
        // finished <--- queue
        if !queue.isEmpty && time - queue[0][1] == bridge_length {
            finished.append(queue.removeFirst())
        }

        // queue <--- waits
        if let first = waits.first {
            if queue.map({ $0[0] }).reduce(0, +) + first <= weight {
                queue.append([waits.removeFirst(), time])
            }
        }

        time += 1
    }     
    return time 
}