백준 11501 주식 파이썬

문제

홍준이는 요즘 주식에 빠져있다. 그는 미래를 내다보는 눈이 뛰어나, 날 별로 주가를 예상하고 언제나 그게 맞아떨어진다. 매일 그는 아래 세 가지 중 한 행동을 한다.
1. 주식 하나를 산다.
2. 원하는 만큼 가지고 있는 주식을 판다.
3. 아무것도 안한다.
홍준이는 미래를 예상하는 뛰어난 안목을 가졌지만, 어떻게 해야 자신이 최대 이익을 얻을 수 있는지 모른다. 따라서 당신에게 날 별로 주식의 가격을 알려주었을 때, 최대 이익이 얼마나 되는지 계산을 해달라고 부탁했다.
예를 들어 날 수가 3일이고 날 별로 주가가 10, 7, 6일 때, 주가가 계속 감소하므로 최대 이익은 0이 된다. 그러나 만약 날 별로 주가가 3, 5, 9일 때는 처음 두 날에 주식을 하나씩 사고, 마지막날 다 팔아 버리면 이익이 10이 된다.

입력

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

출력

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

코드

# 주식
import sys

def cal(days):
    result = 0
    # 최고가 위치를 찾고, 그 전에 있는 날에 매일매일 주식을 하나씩 사면 되는 거 아닐까?
    # 근데 이제 최고가 이후에도 음...문제가 있네
    while len(days) > 0:
        high = max(days)
        idx = 0
        r_days = days[::-1]
        for i in range(len(days)):
            if r_days[i] == high:
                idx = len(days) - i
                break

        for i in range(0, idx, 1):
            if days[i] < high:
                result += high - days[i]

        days = days[idx:]

    return result

case_num = int(sys.stdin.readline().rstrip())

result = []
for i in range(case_num):
    _ = int(sys.stdin.readline().rstrip())
    days = list(map(int, sys.stdin.readline().rstrip().split(' ')))
    #여기서 계산하고 바로 프린트 -> 함수로 만들자
    result.append(cal(days))

for i in result:
    print(i)

문제를 풀면서 느낀 문제점

이번 문제는 좀 흔한 문제라서 쉽게 풀 수 있었다.
처음에는 최고가를 찾은 후 그 뒤를 찾는 코드를 작성하지 않아서 틀렸는데, while 함수 안에 계산을 넣고 days를 슬라이싱하여서 해결하였다.