[SWEA/Python][D2]1859. 백만장자 프로젝트

<문제>

25년 간의 수행 끝에 원재는 미래를 보는 능력을 갖게 되었다. 이 능력으로 원재는 사재기를 하려고 한다.

다만 당국의 감시가 심해 한 번에 많은 양을 사재기 할 수 없다.

다음과 같은 조건 하에서 사재기를 하여 최대한의 이득을 얻도록 도와주자.

    1. 원재는 연속된 N일 동안의 물건의 매매가를 예측하여 알고 있다.
    2. 당국의 감시망에 걸리지 않기 위해 하루에 최대 1만큼 구입할 수 있다.
    3. 판매는 얼마든지 할 수 있다.

예를 들어 3일 동안의 매매가가 1, 2, 3 이라면 처음 두 날에 원료를 구매하여 마지막 날에 팔면 3의 이익을 얻을 수 있다.

 

[입력]

첫 번째 줄에 테스트 케이스의 수 T가 주어진다.

각 테스트 케이스 별로 첫 줄에는 자연수 N(2 ≤ N ≤ 1,000,000)이 주어지고,

둘째 줄에는 각 날의 매매가를 나타내는 N개의 자연수들이 공백으로 구분되어 순서대로 주어진다.

각 날의 매매가는 10,000이하이다.


[출력]

각 테스트 케이스마다 ‘#x’(x는 테스트케이스 번호를 의미하며 1부터 시작한다)를 출력하고, 최대 이익을 출력한다.

 

<나의 풀이>

T = int(input())
for tc in range(1, T + 1):
    N = int(input())
    data = list(map(int,input().split()))

    ans = 0
    max_cost = data[-1]
    for i in range(N-2, -1, -1):
        if max_cost > data[i]:
            ans += max_cost - data[i]
        else:
            max_cost = data[i]
    print('#{} {}'.format(tc, ans))

 

<REVIEW>

이 문제는 비교적 쉽게 풀었다. for문을 앞에서 차례대로 푸는 게 아닌 뒤에서 차례대로 접근했고, 여기서 핵심은 else일 때, max_cost를 data[i]로 지정해 다시 거기서부터 시작하는것인 것 같다. 한 데이터에서 숫자 2개를 비교해야 해서 처음에는 2개의 for문을 돌려야 하는건가 싶었다. 하지만 for문 2개를 이용하니 쓸데없는 값까지 같이 더해져서 틀렸다고 생각했고, 어찌됐든 하나의 값이 지정이 되어서 for문으로 나머지들을 비교를 해야했기에 max_cost변수를 사용했다. 생각보다 간단했다. 

 

<출처>https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=2&contestProbId=AV5LrsUaDxcDFAXc&categoryId=AV5LrsUaDxcDFAXc&categoryType=CODE&problemTitle=&orderBy=FIRST_REG_DATETIME&selectCodeLang=ALL&select-1=2&pageSize=10&pageIndex=1