백준 11501: 주식

2022. 4. 10. 22:20알고리즘/BFS 알고리즘

문제

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

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

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

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

입력

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

출력

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

 


내 제출

public class No11501 {
    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int num = Integer.parseInt(br.readLine());
        StringBuffer sb = new StringBuffer();

        for (int i = 0; i < num; i++) {
            int day = Integer.parseInt(br.readLine());
            int[] arr = new int[day];
            StringTokenizer st = new StringTokenizer(br.readLine());
            Stack<Integer> s = new Stack<>();
            for (int j = 0; j < day; j++) {
                arr[j] = Integer.parseInt(st.nextToken(" "));
            }

            int max = 0;
            long sum = 0;
            for (int j = day - 1; j >= 0; j--) {
                if(arr[j]> max){
                    max=  arr[j];
                }else{
                    sum += (max-arr[j]);
                }
            }


            System.out.println(sum);


        }

    }
}

각 날들의 주가를 배열에 담은 뒤, 뒤에서 부터 고점을 찾아나선다.
만약 신 고점이 등장하면(뒤에서 부터 조사하고있다!! 빠른 인덱스의 요소가 더 크다면) 고점을 갱신한다.
현재 고점보다 작다면, 현재의 고점 - 현재 가치를 sum에 더해주는 방식이다.

'알고리즘 > BFS 알고리즘' 카테고리의 다른 글

백준 4963: 섬의 개수  (0) 2022.04.26
백준 2667번: 단지번호붙이기  (0) 2022.04.10
백준 2583번: 영역구하기  (0) 2022.04.09
백준 7562: 나이트의 이동  (0) 2022.04.08
백준 1012번: 유기농 배추  (0) 2022.04.06