[BOJ] 백준 11501번 주식

백준 11501번 주식 (실버2)

문제

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

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

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

문제 풀이

1. 그리디 알고리즘에 맞게 가장 큰 값을 찾고, 그 값 까지의 이익을 계산.
   이후 계산한 주식들을 제외하고 또 계산해야 할 주식들이 남아있다면 다시 가장 큰 값을 찾고 이익 계산하기 반복.
   -> 91% 에서 시간초과...

   T = int(input())
   for _ in range(T):
       N = int(input())
       data = list(map(int, input().split()))
   buy = 0
   while data:
       max_value = max(data)
       index = data.index(max_value)
       # print(max_value, "/", index)
       buy += max_value * index - sum(data[:index])
   
       if index < len(data) - 2:
           data = data[index + 1 :]
       else:
           break
   print(buy)

2. 앞에서부터 계산하지 말고, 뒤에서 부터 계산하자.
   앞에서부터 접근하면 반복문을 순회하며 지나온 개수를 더하고, 다음 인덱스와 비교를 하지만,
   뒤에서 부터 접근한다면 단순히 현재 max값보다 현재 인덱스 값이 더 작다면 그 차이만큼 이익을 더하면 된다.

   import sys
   
   T = int(sys.stdin.readline().strip())
   for _ in range(T):
       N = int(sys.stdin.readline().strip())
       data = list(map(int, sys.stdin.readline().split()))
       result = 0
       max = 0
       for i in range(len(data) - 1, -1, -1):
           if data[i] > max:
               max = data[i]
           else:
               result += max - data[i]
       print(result)