백준 11501번: 주식 (파이썬)

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

11501번: 주식

 

문제

 

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

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

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

 

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

 

입력

 

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

 

출력

 

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

 

풀이

 

최대의 이익을 보기 위해서는 가장 비쌀 때 가장 많이 팔 수 있어야 한다.

앞에서부터 탐색했을 때는 반복되는 부분이 많아 시간 초과 문제가 발생하였다.

 

뒤에서부터 탐색을 하며 당일의 주가(price 변수)가 당일 기준 미래의 주가 중 가장 높은 가격(max_price 변수)보다 낮거나 같다면 주식을 구매하도록 한다.

이는 총 이익을 나타내는 balance 변수에서 지출만큼 빼주고, 주식의 개수를 나타내는 stock 변수를 증가시킴으로써 구현하였다.

만약 당일의 주가가 당일 기준 미래의 주가 중 가장 높은 가격보다 높다면, 당일 기준 미래의 모든 주식을 미래의 주가 중 가장 높은 가격에 판매하도록 하고, 새로운 최고가를 갱신한다.

이 과정을 첫 날까지 반복하도록 한다.

 

소스 코드

 

import sys

t = int(sys.stdin.readline())
for _ in range(t):
    n = int(sys.stdin.readline())
    prices = list(map(int, sys.stdin.readline().split()))

    balance = 0
    stock = 0
    max_price = 0
    while prices:
        price = prices.pop()
        if price > max_price:
            balance += stock*max_price
            stock = 0
            max_price = price
        else:
            balance -= price
            stock += 1
    balance += stock*max_price
    stock = 0

    print(balance)