프로그래머스 150369. 택배 배달과 수거하기

https://school.programmers.co.kr/learn/courses/30/lessons/150369

프로그래머스

def solution(cap, n, deliveries, pickups):
    ans = 0
    delivery_remain, pickup_remain = 0, 0

    for i in range(n - 1, -1, -1):  # 가장 먼 집부터 역순으로 처리
        delivery_remain += deliveries[i] # 처리할 작업 누적
        pickup_remain += pickups[i]
		
        # 한 집의 두 작업을 모두 완료할 때까지 반복
        while delivery_remain > 0 or pickup_remain > 0:
            ans += (i + 1) * 2  # 왕복 거리
            delivery_remain -= cap  # 배달 처리
            pickup_remain -= cap  # 수거 처리

    return ans