백준: 11501번 주식 [그리디 알고리즘, 실버 2]

문제

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

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

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

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

입력

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

출력

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

예제 입력 1 복사

3
3
10 7 6
3
3 5 9
5
1 1 3 1 2

예제 출력 1 복사

0
10
5

 

문제풀이

: 주식을 사고 팔 시 최대 이익을 구함 -> 그리디 알고리즘 사용

반복문을 뒤에서 부터 접근했을 때 쉽게 풀 수 있다.

앞에서 부터 접근하게 되면, 반복문을 순회하면서 지나온 개수를 더하고, 다음 인덱스에서 가격이 떨어질 때 팔아야해서 복잡하다.

그러나 뒤에서 부터 접그낳면 단순히 현재 Max 값보다 현재 인덱스 값이 더 작다면 그 차이만큼 이익을 더하면 된다.

 

import sys

T = int(sys.stdin.readline().rstrip("\n")) # 테스트 케이스 수 T

for _ in range(T): 
    N = int(sys.stdin.readline().rstrip("\n")) # 날의 수를 나타내는 자연수 N
    cycle = list(map(int, sys.stdin.readline().rstrip("\n").split())) # 날별 주가

    answer = 0 # 최대 이익 결과
    max_value = 0 # max 값

    for i in range(len(cycle)-1, -1, -1): # 뒤에서 부터 반복문 실행!
        if cycle[i] > max_value: # 현재 max값보다 현재 인덱스값이 클 때 max 값 업데이트
            max_value = cycle[i]
        else: # 현재 max값보다 현재 인덱스값이 작을 때 최대 이익 구함
            answer += max_value - cycle[i]

    print(answer)