[Programmers] 택배 배달과 수거하기

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

프로그래머스

 

문제 해결

처음에는 이걸 deliveries를 한 번 훑고 난 다음에 pickups을 또 다시 훑어야되나 고민이 되었는데,

예시를 보다보니 힌트를 얻었다.

거리가 가장 먼 곳부터 쭉 보면서 계산을 해주는 방식으로 greedy로 풀었더니

간단하게 풀 수 있었다. 

while문은 택배를 한 번에 수거하지 못한 경우에 cnt를 늘려주면서 거리를 계산해주는 코드이다.

 

코드

def solution(cap, n, deliveries, pickups):
    answer = 0
    deliver, pickup = 0, 0
    
    for i in range(n-1, -1, -1):
        if deliveries[i] != 0 or pickups[i] != 0:
            cnt = 0
            while (deliver < deliveries[i] or pickup < pickups[i]):
                cnt += 1
                deliver += cap
                pickup += cap
            
            deliver -= deliveries[i]
            pickup -= pickups[i]
            answer += ((i+1) * cnt * 2)
            
    return answer