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

 

---

 

풀이

바킹독 선생님의 그리디 연습문제를 풀다가 접한 문제. 그리디 목록에서 보고 푸는거다 보니, 이 문제를 그리디로 푸는걸 알고 있는 상태였다. 그래서 속으로 '그리디인걸 알면 구현만 잘 하면 돼지!' 라고 생각했는데, 20분동안 봐도 풀이 방법이 전혀 생각 안 나더라. 

주식 가격을 맨 뒤 인덱스부터 확인한다. 맨 뒤의 주식 가격을 tempMax에 저장한 후, 인덱스 i를 1씩 줄여나가면서

1. i의 주식 가격이 tempMax보다 작다: i때 주식을 사서 현재 주식 가격으로 팔면 최대 이득이 된다. 왜냐하면, 뒤에서부터 샀으니까 tempMax는 i부터 N-1번째 주식 중 가장 비싼 가격이 된다.
2. idx의 주식 가격이 tempMax보다 크다: tempMax를 i의 주식 가격으로 바꿔준다.

 

배운 점

이 문제를 다시 보면 어떻게 접근할까?

1. N이 백만이다: O(N^2) 이상의 알고리즘은 얄짤없음, O(NlogN) (분할정복 등의 느낌) 정도로 전체를 체크할 수는 없을 듯하다 -> O(N) 알고리즘이 필요하며, 이런 건 그리디다라고 인지하기
2. 20분동안 앞에서부터 바라봤는데, 뒤에서부터 바라보면 문제가 쉽게 풀리더라. 보는 관점을 다양하게 해야하는데, 결국 문제 양치기부터 해야하지 않을까 싶다.

 

코드

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

public class Main {

	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int T = Integer.parseInt(br.readLine());
		StringTokenizer st;
		for (int t = 0; t < T; t++) {
			int N = Integer.parseInt(br.readLine());
			long[] stockPrice = new long[N];
			st = new StringTokenizer(br.readLine());
			for (int i = 0; i < N; i++) {
				stockPrice[i] = Long.parseLong(st.nextToken());
			}
			long profitMax = 0;
			
			long tempMax = stockPrice[N-1];
			for (int i = N-1; i >= 0; i--) {
				if (tempMax < stockPrice[i]) tempMax = stockPrice[i];
				else profitMax += (tempMax - stockPrice[i]);
			}
			System.out.println(profitMax);
		}
	}

}