[Baekjoon] 11501번 주식

문제

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

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

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

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

입력

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

출력

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

예제

Example 1:

Input: 
3
3
10 7 6
3
3 5 9
5
1 1 3 1 2
Output: 
0
10
5

조건

시간 제한 : 5초

메모리 제한 : 256 MB

풀이과정

잘못된 풀이

계속 91%에서 시간초과가 나서 답답했다. n이 최대 1000000개라서 n번 돌때 매번 최댓값을 갱신하는 나의 코드로는 $O(N^2)$의 시간복잡도를 가지게 되므로 시간초과가 당연하다. 다른 풀이법이 필요해보인다.

import sys
input = sys.stdin.readline

# t = 테스트케이스
t = int(input())

for _ in range(t):  

    # n = 날의 수
    n = int(input())

    # 주가
    price = list(map(int, input().split()))

    stock = []
    ans = 0
    index = 0
    while(1):
  
        last = max(price[index:])

        # 주식 팔아버리기
        if price[index] == last:
            while stock:
                ans += last - stock.pop()
            index += 1
        elif price[index] < last:
            while(price[index] != last):
                stock.append(price[index])
                index += 1

        if index == n:
            break
    print(ans)

맞는 풀이

해당 글을 보고 풀이방법을 고쳤다. 앞에서풀면 지금 가리키는 수가 주식에서 산 것보다 크더라도 더 뒤에 수가 크다면 다시 계산해야 한다. 따라서 뒤에서 부터 하나씩 빼가면서 탐색을 한다. 값이 커지는 시점은 최댓값이 갱신되는 때이고 값이작아지는 시점은 주식을 파는 때이다.

# t = 테스트케이스
t = int(input())

for _ in range(t):  

    # n = 날의 수
    n = int(input())

    # 주가
    price = list(map(int, input().split()))
    last = price[n-1]
    stock = []
    ans = 0
    index = 0
    for i in range(n-1,-1,-1):
        if price[i] < last:
            ans += last-price[i]
        else:
            last = price[i]
    print(ans)