BOJ-11501 주식 Swift

문제

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

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

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

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

입력

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

출력

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

내가 푼 풀이

- 주가가 최대일때 팔면 최대이익이 생긴다.

- 이는 날이 지날수록 다음날이 이전날보다 주가가 높으면 매수한다.

- 이 방법으로 매수해서 최고점을찍어 다음날이 이전날보다 주가가 떨어질때, 매수했던 주식을 판매하는 방법을 이용했는데, 자꾸 틀렸다고 떴다.

- 아마 출력조건에서 부호있는 64bit 정수형으로 표현해야하고, 반례가 있어서 다른 방법을 이용했다.

- 오늘이 다음날보다 주가가 낮으면 무조건 주식을 사면되는데, 반대로 최고가를 찍은 날을 알고있다면 그 전날의 주식들이 최고가보다 낮다면 모두 차익을 남기고 팔면된다.

- 이는 주어진 주가를 배열로 입력하고, 역순으로 주식의 최고가를 갱신하여 이전날이 주가가 낮다면 차익이 생긴다는 얘기다.

 

import Foundation

// 테스트케이스 개수
let cc = Int(readLine()!)!

for i in 0..<cc {
    let count = Int(readLine()!)!
    var arr = readLine()!.split(separator: " ").map{ UInt64($0)! }
    var max: UInt64 = 0
    var result: UInt64 = 0
    
    // 역순으로 주가의 최고가보다 전날이 적으면 차익을 계속 더해준다.
    for j in (0..<count).reversed() {
        if arr[j] > max {
            max = arr[j]
        } else {
            result += (max - arr[j])
        }
    }
    print(result)
}