[백준/Python] 11501 - 주식

문제

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

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

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

입력

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

출력

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

풀이
예전에 알고리즘 수업에서 풀었던 문제랑 매우 유사한데, 너무 오래돼서 풀이가 잘 기억안났다
어쨌거나 최고의 이익을 얻기 위해서는 주식이 올랐을 때 보유 주식을 팔면되는데, 주의할 점은 주식이 계속해서 오른다면 매일 주식을 파는 것이 아니라 주식 가격이 가장 높은 날 주식들을 팔아야한다는 것이다
이 점 때문에 주어진 순서대로 loop를 돌며 풀기는 좀 까다로운 문제였다
따라서 주어진 리스트를 reverse하여 최대값을 갱신해가면서 값이 떨어졌을 때 매도한다
3, 5, 9 ==> 9, 5, 3

코드

import sys

t = int(sys.stdin.readline())

for _ in range(t):
    day = int(sys.stdin.readline())
    arr = list(map(int, sys.stdin.readline().split()))
    arr.reverse()

    max_val = arr[0]
    benefit = 0
    for elem in arr[1:]:
        if max_val > elem:
            benefit += max_val-elem
        else:
            max_val = elem
    print(benefit)