코딩테스트 \파이썬\자바/SWEA D2

[SWEA][D2] 1859. 백만 장자 프로젝트 D2 파이썬

익명의 신디 2020. 3. 9. 20:23

📌 문제출처

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV5LrsUaDxcDFAXc&categoryId=AV5LrsUaDxcDFAXc&categoryType=CODE

📖1859. 백만 장자 프로젝트 D2 파이썬

 

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

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

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

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

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


[입력]

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

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

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

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


[출력]

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


[예제 풀이]

1번째 케이스는 아무 것도 사지 않는 것이 최대 이익이다.

2번째 케이스는 1,2일에 각각 한 개씩 사서 세 번째 날에 두 개를 팔면 10의 이익을 얻을 수 있다.

 

 

💡<PASS>

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
# 백만장자 D2 파이썬
 
= int(input())
for t in range(1,T+1):
    num = int(input())
    arr = list(map(int,input().split()))
    last = arr[-1]
    cnt = 0
    for i in range(len(arr)-1,-1,-1): # 핵심! 뒤부터 보기
        if last > arr[i]:
            cnt += last-arr[i]
        else# 같거나 작을 때 
            last = arr[i]
    print("#{} {}".format(t, cnt))
 
 

 

👀 후기 ) 

배열 뒤 부터 탐색하면 반목문 하나로 끝낼 수 있습니다.
배열의 뒤부터 큰 값을 찾고, 그 뒤 더 큰 값이 나올 떄 까지 누적시키면 사재기의 이익을 구할 수 있다.

 

미래를 보는 능력처럼 우리도 모든 값을 알 고 있기 때문에 가능했던 문제..

그러나 이 반대로의 생각이 매우 어렵다 ㅠ