[백준] 11501 주식 (C++)

문제

홍준이는 요즘 주식에 빠져있다. 그는 미래를 내다보는 눈이 뛰어나, 날 별로 주가를 예상하고 언제나 그게 맞아떨어진다. 매일 그는 아래 세 가지 중 한 행동을 한다.

1. 주식 하나를 산다.
2. 원하는 만큼 가지고 있는 주식을 판다.
3. 아무것도 안한다.

홍준이는 미래를 예상하는 뛰어난 안목을 가졌지만, 어떻게 해야 자신이 최대 이익을 얻을 수 있는지 모른다. 따라서 당신에게 날 별로 주식의 가격을 알려주었을 때, 최대 이익이 얼마나 되는지 계산을 해달라고 부탁했다.

예를 들어 날 수가 3일이고 날 별로 주가가 10, 7, 6일 때, 주가가 계속 감소하므로 최대 이익은 0이 된다. 그러나 만약 날 별로 주가가 3, 5, 9일 때는 처음 두 날에 주식을 하나씩 사고, 마지막날 다 팔아 버리면 이익이 10이 된다.

입력

입력의 첫 줄에는 테스트케이스 수를 나타내는 자연수 T가 주어진다. 각 테스트케이스 별로 첫 줄에는 날의 수를 나타내는 자연수 N(2 ≤ N ≤ 1,000,000)이 주어지고, 둘째 줄에는 날 별 주가를 나타내는 N개의 자연수들이 공백으로 구분되어 순서대로 주어진다. 날 별 주가는 10,000이하다.

출력

각 테스트케이스 별로 최대 이익을 나타내는 정수 하나를 출력한다. 답은 부호있는 64bit 정수형으로 표현 가능하다.

---

풀이

먼저 행동에서 "원하는 만큼 가지고 있는 주식을 판다"에 따라 주식을 한 개만 팔 수 있는 것이 아니라 여러 개 팔 수 있다.

그러므로, 미래에 오늘의 주가보다 1이라도 더 비싼 주가인 날이 존재한다면 오늘은 반드시 주식을 구매해야 한다.

이에 따라 '주가가 가장 비쌀 때의 가격'에 집중하여 마지막에서부터 처음으로 이동하면 쉽게 답을 구할 수 있다.

 

예제 입력 1의 마지막 테스트 케이스를 보면 다음과 같다.

 

그리고 주의할 점은 답이 64비트 정수형이기 때문에 이익을 long long 등의 자료형으로 저장해야 한다.

코드

#include <iostream>
using namespace std;
int prices[1000001];

long long solve(int n){
    // 마지막날부터 이동하면서 제일 비싼 가격을 저장
    int maxPrice = prices[n-1];
    long long answer = 0;
    for (int i = n-2; i >= 0; i--){
        // 전날 가격이 더 높거나 같은 경우는 살 필요가 없음
        if (prices[i] >= maxPrice){
            maxPrice = prices[i];
        }
        // 전날 가격이 더 낮다면 판매
        else {
            answer += (maxPrice - prices[i]);
        }
    }
    return answer;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);

    int t;
    cin >> t;

    while (t--){
        int n;
        cin >> n;

        for (int i = 0; i < n; i++){
            cin >> prices[i];
        }

        cout << solve(n) << "\n";
    }
}