[SW Expert] 1859번_백만장자프로젝트

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

SW Expert Academy

처음엔 0번째 인덱스부터 시작하여 최대를 찾고 어쩌고의 방법을 생각해봤으나 누가봐도 시간초과...

(안 날 수도 있지만 꽤나 긴 코드를 짜야할 것이다.)

배열의 끝에서 접근할 생각만 해낸다면 어렵지 않게 풀 수 있는 문제였다. 

for t in range(int(input())):
    d = int(input())
    days = list(map(int,input().split()))
    
    cnt = 0; accum = 0; ans=0; price = days[-1]
    for day in days[-2::-1]:
        if day<price:
            cnt += 1
            accum += day
        else:
            ans += price*cnt - accum
            price = day
            cnt = 0 ; accum = 0
    if cnt>=1:
        ans += price*cnt - accum
                
    print("#{0} {1}".format(t+1,ans))

변수 설명을 덧붙이자면 price는 팔 가격, cnt는 파는 물건의 개수, accum은 파는 물건의 가격 누적값이다.