[Algorithm] 다리를 지나는 트럭

🖥 문제

트럭 여러 대가 강을 가로지르는 일차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 다리에는 트럭이 최대 bridge_length대 올라갈 수 있으며, 다리는 weight 이하까지의 무게를 견딜 수 있습니다. 단, 다리에 완전히 오르지 않은 트럭의 무게는 무시합니다.

solution 함수의 매개변수로 다리에 올라갈 수 있는 트럭 수 bridge_length, 다리가 견딜 수 있는 무게 weight, 트럭 별 무게 truck_weights가 주어집니다. 이때 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 return 하도록 solution 함수를 완성하세요.

 

🤔 풀이 방법

• 다리에 올라갈 수 있는 트럭 수 : 2 / 다리가 견딜 수 있는 무게 : 10, / 트럭 별 무게 : [7, 4, 5, 6] 

  경과 시간	다리를 지난 트럭	다리를 건너는 트럭	대기 트럭
  0			7, 4, 5, 6
  1		7	4, 5, 6
  2		7	4, 5, 6
  3	7	4	5, 6
  4	7	4, 5	6
  5	7, 4	5	6
  6	7, 4, 5	6
  7	7, 4, 5	6
  8	7, 4, 5, 6

 

• 다리에 올라갈 수 있는 트럭 수 : 100 / 다리가 견딜 수 있는 무게 : 100, / 트럭 별 무게 : [10]

  경과 시간	다리를 지난 트럭	다리를 건너는 트럭	대기 트럭
  0			10
  1 ~ 100		10
  101	10

 

👩🏻‍💻 Swift Code

import Foundation

func solution(_ bridge_length:Int, _ weight:Int, _ truck_weights:[Int]) -> Int {
    var truck = truck_weights
    var bridge: [Int] = Array(repeating: 0, count: bridge_length)
    var arrived: [Int] = []
    var weightSum = 0
    var time = 0


    while true {
        if arrived == truck_weights { break }

        time += 1

        if bridge.first! != 0 {
            arrived.append(bridge.first!)
        }

        weightSum -= bridge.first!
        bridge.removeFirst()

        if truck != [] && weightSum + truck.first! <= weight {
            weightSum += truck.first!
            bridge.append(truck.first!)
            truck.removeFirst()
        } else {
            bridge.append(0)
        }

    }

    return time
}

 

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