(Python) - BOJ(11501번) : 주식

https://www.acmicpc.net/problem/11501

11501번: 주식

---

• 해설 : 

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

 

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

 

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

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

 

 

 

---

• 풀이 :

그리디하게 해결할 수 있다. 마지막날의 값을 비교할 대상으로 놓고, 마지막날부터 역으로 진행하며 그 날의 가격이 더 낮다면 그 차이만큼 이득을 볼 수 있는 것이므로 답에 덥해준다. 아니라면 비교할 대상(가격의 max)를 그 값으로 바꾸어 이득을 최대로 챙길 수 있는 방법을 찾는다.

 

 

 

---

• 소스코드 : 

 

if __name__ == "__main__":
    T = int(input())
    for i in range(T):
        N = int(input())
        nums = list(map(int,input().split()))
        curr = nums[-1]
        answer = 0
        for j in range(N-1,-1,-1):
            if nums[j] < curr:
                answer += curr-nums[j]
            else:
                curr = nums[j]
        print(answer)