[BOJ][⚪2][백준#11501] 주식

문제 출처

BAEKJOON Online Judge #11501

문제

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

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

홍준이는 미래를 예상하는 뛰어난 안목을 가졌지만, 어떻게 해야 자신이 최대 이익을 얻을 수 있는지 모른다. 따라서 당신에게 날 별로 주식의 가격을 알려주었을 때, 최대 이익이 얼마나 되는지 계산을 해달라고 부탁했다. 예를 들어 날 수가 3일이고 날 별로 주가가 10, 7, 6일 때, 주가가 계속 감소하므로 최대 이익은 0이 된다. 그러나 만약 날 별로 주가가 3, 5, 9일 때는 처음 두 날에 주식을 하나씩 사고, 마지막날 다 팔아 버리면 이익이 10이 된다.

입력

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

출력

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

예제

입력

3
3
10 7 6
3
3 5 9
5
1 1 3 1 2

출력

0
10
5

My Sol

const fs = require("fs");
const input = fs
  .readFileSync('/dev/stdin')
  .toString()
  .split("\n")
  .map((a) => a.trim());

const foo = (i) => {
  const N = +input[2 * i + 1];
  const arr = input[2 * i + 2].split(" ").map(Number);
  let k = N - 1;
  let maxx = arr[k];
  let ssum = 0;
  while (k >= 0) {
    if (maxx < arr[k]) {
      maxx = arr[k];
    } else {
      ssum += maxx - arr[k];
    }
    k--;
  }
  return ssum;
};

const T = +input[0];
for (let i = 0; i < T; i++) {
  console.log(foo(i));
}

뒤에서부터 풀어낸다.

1. 가장 마지막의 수를 maxx로 둔다.
2. 뒤에서부터 앞의 날로 오면서, maxx보다 작거나 같으면 그 차를 ssum에 더해준다.
3. 만약 maxx보다 큰 값을 발견하면 maxx를 해당 값으로 둔다.
4. 즉, maxx보다 큰 값을 발견하기 전까지의 구간에서는 maxx가 가장 마지막값이면서도 가장 큰 값이므로 해당 구간에서는 계속 주식을 사모은다고 생각하면 된다.
5. 그리고 값이 maxx인 날에 한 번에 팔아버리는 것이다.
6. 이렇게 0의 인덱스까지 오게 되면 ssum을 출력한다.

결과

맞았습니다!!