[Silver II] 주식 - 11501

문제 링크

성능 요약

메모리: 331592 KB, 시간: 1192 ms

분류

그리디 알고리즘(greedy)

문제 설명

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

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

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

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

입력

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

출력

각 테스트케이스 별로 최대 이익을 나타내는 정수 하나를 출력한다. 답은 부호있는 64bit 정수형으로 표현 가능하다.

문제 해결 아이디어

        1. prices 배열의 가장 뒷 부분부터 탐색합니다.
        2. rightPriceMax는 최고점을 저장합니다.
        3.  만약 현재 주가 Price >= rightPriceMax 같으면, rightPriceMax = price저장합니다,
            그 외의 경우 최고점 - 현재 주가 = Profit을 maxProfit에 더해줍니다.
        4. 배열을 모두 탐색할때까지 2번 ~ 3번을 반복합니다.

    Data Type
        long maxProfit <<
        [1,1,1 ...... 10,000]경우 최대 프로핏은 10,000,000,000 > 2^31 - 1보다 크기 때문에 Long Type으로 해준다.
    Algorithm
        투포인터

    Time Complexity
        O(N): 10 ^ 6 => 0.01초

솔루션 Github

import java.io.*;
import java.util.*;

public class Main {
    /*
        Overview

        Solution
            1. prices 배열의 가장 뒷 부분부터 탐색합니다.
            2. rightPriceMax는 최고점을 저장합니다.
            3.  만약 현재 주가 Price >= rightPriceMax 같으면, rightPriceMax = price저장합니다,
                그 외의 경우 최고점 - 현재 주가 = Profit을 maxProfit에 더해줍니다.
            4. 배열을 모두 탐색할때까지 2번 ~ 3번을 반복합니다.

        Data Type
            long maxProfit <<
            [1,1,1 ...... 10,000]경우 최대 프로핏은 10,000,000,000 > 2^31 - 1보다 크기 때문에 Long Type으로 해준다.
        Algorithm
            투포인터

        Time Complexity
            O(N): 10 ^ 6 => 0.01초
     */

    public static void solution(List<List<Integer>> priceCharts){
        StringBuilder maxProfits = new StringBuilder();

        for (List<Integer> prices: priceCharts) {
            long maxProfit = 0;
            long rightPriceMax = -1; // 최고점을 저장합니다.

            for(int i = prices.size() - 1; i > -1; i--){
                int price = prices.get(i);
                if(rightPriceMax <= price){
                    rightPriceMax = price;
                }
                else {
                    maxProfit += rightPriceMax - price;
                }
            }
            maxProfits.append(maxProfit).append("\n");
        }

        System.out.print(maxProfits.deleteCharAt(maxProfits.length() - 1));
    }

    public static void main(String[] args) {
        try{
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            StringTokenizer st = new StringTokenizer(br.readLine());
            int T = Integer.parseInt(st.nextToken());
            List<List<Integer>> priceCharts = new ArrayList<>();

            for(int i = 0; i < T; i++){
                st = new StringTokenizer(br.readLine());
                List<Integer> chart = new ArrayList<>();
                int N = Integer.parseInt(st.nextToken());
                st = new StringTokenizer(br.readLine());

                for(int j = 0; j < N; j++){
                    chart.add(Integer.parseInt(st.nextToken()));
                }

                priceCharts.add(chart);
            }

            solution(priceCharts);
        }
        catch (Exception ex){
            System.out.println(ex);
        }
    }
}