프로그래머스 택배 배달과 수거하기 (C++)

안녕하세요~

오늘은 프로그래머스 택배 배달과 수거하기 문제를 풀어봐요!

 

분명히 LV.2인데 풀이법이 바로 안떠오르더라구요ㅠㅠ!!!

그래서 아래 분이 푸신 설명보고 이해하면서 풀었습니다.

Greedy하게 풀어야하더라구용

 

https://school.programmers.co.kr/questions/43364

프로그래머스

 

 

#include <string>
#include <vector>

using namespace std;

long long solution(int cap, int n, vector<int> deliveries, vector<int> pickups) {
    long long answer = 0;
    
    int d = 0;
    int p = 0;
    int cnt = 0;
    for(int i = n - 1; i >= 0; i--)
    {
        cnt = 0;
        d -= deliveries[i];
        p -= pickups[i];
        
        while(d < 0 || p < 0)
        {
            d += cap;
            p += cap;
            
            cnt++;
        }
        
        answer += (i + 1) * 2 * cnt;
    }
    
    return answer;
}