[백준] #11501 주식 - JAVA

❔ 문제

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

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

🔍 문제 풀이

이 문제는 미래의 주가를 정확히 예측할 수 있을 때, 최대 이익을 구하는 알고리즘을 구해야하는 문제이다.
이익을 최대화하기 위해, 앞으로 주가가 가장 높은 날에 팔면된다.
즉, 각 날의 주가를 확인하면서, 뒤에서부터 최댓값을 갱신하며 최적의 선택을 수행할 수 있으므로 그리디 알고리즘을 적용할 수 있다.

1. 뒤에서 부터 주가를 확인하며, 현재까지의 최댓값을 저장한다.

• 주가는 항상 최댓값에서 낮은 가격일 때 구매하는 것이 이득이므로, 최대 주가를 계속 갱신해 준다.

2. 현재 주가가 최댓갓 보다 낮다면, 그 차이만큼 이익을 얻는다.
3. 최댓값을 갱신하며, 모든 날짜를 확인하여 최종 이익을 출력한다.

☑️ 최종 코드

import java.io.*;
public class Main {

	public static void main(String[] args) throws Exception{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int t = Integer.parseInt(br.readLine());
		for(int tc = 0; tc < t; tc ++) {
			int n = Integer.parseInt(br.readLine());
			String[] input = br.readLine().split(" ");
			long result = 0;
			int[] arr = new int[n];
			
			for(int i = 0; i < n; i++) {
				arr[i] = Integer.parseInt(input[i]);
			}
			
			int max = 0;
			for(int i = n - 1; i > -1; i--) {
				if(arr[i] > max) {
					max = arr[i];
				}else {
					result += (max - arr[i]);
				}
				
			}
			
			System.out.println(result);
			
		}
	}

}