[프로그래머스] 택배 배달과 수거하기 (Python)

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

프로그래머스

 

🔥 감이 잘 오지 않아서 조금 헤맸던 문제ㅠㅜ 다른 분들의 코드를 참고하여 이해해보고, 직접 작성해보는 순으로 공부하였다.

🔥 기억해야 할 포인트 ➡️ 최단 거리로 다녀오려면 제일 먼 곳부터 다녀와야 한다!

 

def solution(cap, n, deliveries, pickups):
    answer = 0
    # 배달과 수거 가능 공간
    d, p = 0, 0
    
    for idx in range(n-1, -1, -1):
        # 해당 집에서 배달과 수거 실행
        d += deliveries[idx]
        p += pickups[idx]
        
        while d > 0 or p > 0:
            # d나 p가 양수라면 => 여유공간 없음 => 창고 들렸다 와야 함
            # d나 p가 음수가 될 때까지 => 여유공간이 생길 때까지 왔다갔다
            d -= cap
            p -= cap
            answer += (idx+1)*2
        
    return answer