[Python/파이썬] 백준 11501번 주식

11501번: 주식

 

코드

for _ in range(int(input())):
    n = int(input())
    lst = list(map(int, input().split()))

    highest = -1
    ans = 0
    for i in range(n-1, -1, -1):
        if highest < lst[i]:
            highest = lst[i]
        else:
            ans += (highest-lst[i])
    print(ans)

 

풀이는 간단하다.

날짜를 역순으로 가며 최대 주식 가격인 highest보다 낮은 가격이면 무조건 사들이고, highest보다 높은 가격이 나오면 highest를 그 값으로 수정해주면 된다.

거꾸로 진행하며 이전에 가장 높은 가격보다 낮은 가격은 매입하고, 더 높은 가격이 나오면 highest를 갱신하여 그 앞 날짜의 주식들은 새로운 highest 가격으로 판다고 생각하면 된다.