백준 11501번 : 주식(자바)

https://www.acmicpc.net/problem/11501

문제

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

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

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

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

---

입력

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

---

출력

각 테스트케이스 별로 최대 이익을 나타내는 정수 하나를 출력한다. 답은 부호있는 64bit 정수형으로 표현 가능하다. M개의 줄에 걸쳐 캐릭터의 전투력에 맞는 칭호를 입력 순서대로 출력한다. 어떤 캐릭터의 전투력으로 출력할 수 있는 칭호가 여러 개인 경우 가장 먼저 입력된 칭호 하나만 출력한다.

🔵풀이

이 문제는 N이 1,000,000 이므로 O(n)이하의 시간복잡도가 필요하다고 생각했습니다. 처음엔 Stack을 사용해 해결하려 했지만 잘 되지 않았고 배열을 역순으로 탐색하며 최대이익을 구하였습니다. 최댓값 max를 기준으로 더 큰 값이 나타난다면 max를 변경해주고 새로운 최대값이 나타날때까지 계속해서 max를 기준으로 이익을 계산했습니다.

 

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

class Main {
  public static void main(String[] args) throws IOException {
    BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
    int T = Integer.valueOf(bufferedReader.readLine());

    for (int i=0; i<T; i++){
      int N = Integer.valueOf(bufferedReader.readLine());
      int[] prices = new int[N];
      StringTokenizer stringTokenizer = new StringTokenizer(bufferedReader.readLine());
      int max = Integer.MIN_VALUE;
      long sum = 0;

      for (int j=0; j<N; j++){
        prices[j] = Integer.valueOf(stringTokenizer.nextToken());
      }

      max = prices[N-1];

      for (int j=N-2; j>-1; j--){
        if (max<prices[j]){
          max = prices[j];
        }
        else {
          sum+= max-prices[j];
        }
      }
      System.out.println(sum);
    }
  }
}