[Coding Test][Python][백준] BOJ #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 정수형으로 표현 가능하다.

---

풀이)

※ 풀이 아이디어

Greedy 알고리즘을 통해 역순으로 탐색하면서 가장 높은 주가에서 최대한 팔고 그 전날 까지의 주식을 구매
1. 미래의 가격을 알기 때문에 역순으로 탐색
2. 현재까지 본 날 중 가장 높은 주가 저장
    1. 현재 가격이 최대값보다 작으면 이익
    2. 현재 가격이 최대값보다 크면 최대값 갱신

 

Input = [1, 3, 7, 5, 6, 10, 9, 8, 1] 일 때의 예시입니다.

첫 Iteration에 Max Value가 1로 갱신되고 res에는 변화가 없습니다.

Max value가 8에서 9로 갱신되므로 res에 변화가 없습니다.

이제 최대값 보다 작은 날이 발견되어 res에 값이 더해지게 됩니다.

이 과정을 반복하면 최대 이익을 낼 수 있습니다.

import sys
input = sys.stdin.readline

T = int(input().strip())

for t in range(T):
    N = int(input().strip())

    stock_prices = list(map(int, input().strip().split()))

    res = 0
    max_value = 0

    # Max 값 갱신(뒤에서부터 하기 !)
    # 뒤에서부터 자기보다 큰게 나타나면 Max 값 갱신
    for i in range(N-1, -1, -1):
        if stock_prices[i] > max_value:
            max_value = stock_prices[i]
        else:
            res += max_value - stock_prices[i]
    print(res)