문제
홍준이는 요즘 주식에 빠져있다. 그는 미래를 내다보는 눈이 뛰어나, 날 별로 주가를 예상하고 언제나 그게 맞아떨어진다. 매일 그는 아래 세 가지 중 한 행동을 한다.
- 주식 하나를 산다.
- 원하는 만큼 가지고 있는 주식을 판다.
- 아무것도 안한다.
홍준이는 미래를 예상하는 뛰어난 안목을 가졌지만, 어떻게 해야 자신이 최대 이익을 얻을 수 있는지 모른다. 따라서 당신에게 날 별로 주식의 가격을 알려주었을 때, 최대 이익이 얼마나 되는지 계산을 해달라고 부탁했다.
예를 들어 날 수가 3일이고 날 별로 주가가 10, 7, 6일 때, 주가가 계속 감소하므로 최대 이익은 0이 된다. 그러나 만약 날 별로 주가가 3, 5, 9일 때는 처음 두 날에 주식을 하나씩 사고, 마지막날 다 팔아 버리면 이익이 10이 된다.
입력
입력의 첫 줄에는 테스트케이스 수를 나타내는 자연수 T가 주어진다. 각 테스트케이스 별로 첫 줄에는 날의 수를 나타내는 자연수 N(2 ≤ N ≤ 1,000,000)이 주어지고, 둘째 줄에는 날 별 주가를 나타내는 N개의 자연수들이 공백으로 구분되어 순서대로 주어진다. 날 별 주가는 10,000이하다.
출력
각 테스트케이스 별로 최대 이익을 나타내는 정수 하나를 출력한다. 답은 부호있는 64bit 정수형으로 표현 가능하다.
시간초과 코드
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
t = int(input())
for i in range(t):
d = int(input())
s = list(map(int,input().split()))
asset = []
profit = 0
maxv = max(s)
for j in range(d):
if s[j] != maxv:
asset.append(s[j])
else:
for k in asset:
profit += maxv-k
asset = []
if j+1 < d:
maxv = max(s[j+1:])
print(profit)
91%에서 시간초과가 난다.
아무리 그리디 문제이지만 너무 직관적으로 로직을 만들었는지, 시간 계산을 잘못한 것 같다.
현재 온 지점부터 끝 지점까지 max를 구하고, 만약 현 시점의 값이 그 max보다 작다면 asset 배열에 넣어두었다가, 최대값을 만나면 asset 배열을 순회하면서 다 팔아버리는 방식이다.
하지만 이렇게 되면 배열을 내부에 하나 더 넣어야 하고, max도 매번 비효율적인 방식으로 갱신하므로 시간초과가 났던게 당연하다.
정답코드
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import sys
input = sys.stdin.readline
t = int(input())
for i in range(t):
n = int(input())
s = list(map(int,input().split()))
asset = []
profit = 0
maxv = s[-1]
for j in range(n-2,-1,-1):
if s[j] > maxv:
maxv = s[j]
else:
profit += maxv-s[j]
print(profit)
배열을 거꾸로 순회하기 때문에, 현재 날과 다음날을 비교하는 코드를 굳이 넣지 않아도 되고, 뒤에서부터 최대치를 갱신해나가기 때문에 최대값에 파는 것을 보장할 수 있다.