[알고리즘] 백준 11501 파이썬 - 주식

 

11501번: 주식

문제

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

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

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

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

 

문제풀이

import sys
input = sys.stdin.readline
t=int(input())
#01
for _ in range(t):
    ans=0
    max=0
    a=int(input())
    arr=list(map(int,input().split()))
    #02
    for i in range(len(arr)-1,-1,-1):
        if arr[i]>max:
            max=arr[i]
        else:
            ans+=max-arr[i]
    print(ans)

- #01 : 각각 입력을 a와 arr에 받아준다. 또한 ans와 max라는 변수도 0으로 설정해준다.

- #02 : arr의 길이-1부터 시작하여 0까지 한번에 -1만큼 감소하며 for문을 진행한다.

           쉽게 말해 arr 리스트를 뒤에서 앞으로 받을 때 for문을 이렇게 사용한다.

           조건문을 통해 arr[i] 값이 max보다 크면 max값을 해당 값으로 변경한다.

           그렇지 않다면 ans에 max값과 arr[i]의 차이만큼 더해준다.

           for문이 끝나고 최종적으로 구해진 ans를 출력한다.