[백준 실버2] 11501번: 주식 (그리디) - Java

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 정수형으로 표현 가능하다.

예제 입력 1 

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

예제 출력 1 

0
10
5

---

풀이

이해

문제 한번에 하나씩, 미래를 보고 큰 수가 있다면 산다. 팔 때는 다 팔 수 있으므로 가장 고점에서 현재까지 산 주식들을 다 팔아야 한다. 최댓값을 다시 설정하는 고려가 필요한데, 어디서 할지가 중요하다.

유의해야 했던 테케 (굵은 표시가 파는 지점)

5

1 1 3 1 2

=> 5

6

1 2 4 1 5 6

=> 17

 

=> 뒤에서부터 max를 확인, max보다 작을 때 주식을 사서 차이만큼 이익을 남긴다. max보다 크면 max를 갱신한다. (같을 때는 상관없다)

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.*;

public class Main {
	public void solution() throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		
		int t = Integer.parseInt(br.readLine());
		while(t-- > 0) {
		    int n = Integer.parseInt(br.readLine());
		    int[] stock = new int[n];
		    StringTokenizer st = new StringTokenizer(br.readLine());
		    for(int i=0; i<n; i++) {
		        stock[i] = Integer.parseInt(st.nextToken());
		    }
		    long profit = 0;
		    int max = stock[n-1];
		    for(int i=n-1; i>=0; i--) {
		        if (stock[i] > max)
		            max = stock[i];
		        else
		            profit += (max-stock[i]);
		    }
		    sb.append(profit).append('\n');
		}
		System.out.println(sb);
	}
	public static void main(String[] args) throws Exception {
		new Main().solution();
	}
}