[백준 11501번] 주식 - 파이썬

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

 

💡문제 분석 요약

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

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

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

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

💡알고리즘 설계

주식 정보를 입력받고, 마지막 날짜에서 첫번째 날짜까지 계산한다.

만약 현재 날짜의 주식이 최고가보다 높다면, 현재 날짜의 주식을 최고가로 설정한다.

현재 날짜의 주식이 최고가보다 낮다면, 현재 주식을 구매한 후 미래의 최고가로 판매한다

=>수익금에 최고가 금액에서 현재 주가를 뺀 것을 더한다.

💡코드

import sys
input = sys.stdin.readline

T=int(input())

for _ in range(T):
    N=int(input())
    stock=list(map(int, input().split()))

    money=0
    max=0
    for i in range(N-1, -1, -1):
        if stock[i]>max:
            max=stock[i] #판매할 가격 설정
        else:
            money+=max-stock[i] #판매 (차익: 판매 가격 - 구매 가격)
    print(money)

 

💡 느낀점 or 기억할 정보

처음엔 stock에서 max 값을 매번 구했는데, 시간이 너무 오래걸릴 것 같아 뒤에서부터 풀었다.