import Foundation

func solution(_ cap:Int, _ n:Int, _ deliveries:[Int], _ pickups:[Int]) -> Int64 {
    
    var answer: Int64 = 0
    var d = 0
    var p = 0
    
    for i in stride(from: n - 1, to: -1, by: -1) {
        
        var cnt = 0
        
        d -= deliveries[i]
        p -= pickups[i]
        
        while d < 0 || p < 0 {
            d += cap
            p += cap
            cnt += 1
        }
        
        answer += Int64((i + 1) * 2 * cnt)
    }

    return answer
}

 

'프로그래머스 > Lv.2' 카테고리의 다른 글

[Swift] 연속 부분 수열 합의 개수  (0) 2023.04.25
[Swift] 조이스틱  (0) 2023.04.24
[Swift] 귤 고르기  (0) 2023.03.13
[Swift] 압축  (1) 2023.03.13
[Swift] 주차 요금 계산  (0) 2023.02.07