[Java] 백준 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

 

소스코드

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

public class Main {

	private static BufferedReader br;
	private static StringTokenizer st;
	
	private static int T, N;
	private static long result;
	private static long []days;
	public static void main(String[] args) throws NumberFormatException, IOException {
		br=new BufferedReader(new InputStreamReader(System.in));
		T=Integer.parseInt(br.readLine());
		for(int t=1;t<=T;t++) {
			N=Integer.parseInt(br.readLine());
			days=new long[N];
			result=0;
			st=new StringTokenizer(br.readLine());
			for(int i=0;i<N;i++) {
				days[i]=Integer.parseInt(st.nextToken());
			}
			
			Long max=days[N-1];	// 현재까지 가장 큰 값을 저장
			for(int i=N-2;i>=0;i--) {	// 뒤에서부터 탐색
				if(days[i]>max) {	// 현재 최댓값보다 큰 값이 나오면 최댓값 갱신
					max=days[i];
				} else if(days[i]<max) result+=max-days[i];	// 현재 최댓값보다 작은 값이 나오면 차이만큼 결과에 더해줌
			}
			
			System.out.println(result);
		}
	}

}

1. 뒤에서부터 주가를 탐색하며 현재까지의 최댓값을 max에 저장합니다.

2. 현재 주가보다 큰 값이 나오면 최댓값을 갱신합니다.

3. 현재 주가보다 작은 값이 나오면 최댓값과의 차이만큼 결과에 더해줍니다.