[백준] 주식 - 11501

링크

주식-11501

문제

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

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

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

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

입력

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

출력

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

풀이

이 문제를 풀 때 배열의 처음 인덱스부터 탐색할려다보니 시간초과가 날 가능성이 크게 문제를 풀어나갔었다. 그리고 아이디어를 생각해보니
인덱스를 맨 뒤에서부터 푼다면 문제 접근이 매우 쉬워질 것이라 생각하였다. max값을 찾아서 max값보다 크다면 max=arr(i)로 취해주고
max값보다 크지 않다면 result+=(max-arr(i)) 해당하는 arr 인덱스 값을 max에서 빼주면 끝!

Code

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

public class Main {
	public static void main(String[] args) throws IOException{
		// TODO Auto-generated method stub
		BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
		ArrayList<Long> list=new ArrayList<Long>();
		int t=Integer.parseInt(br.readLine());

		
		for(int i=0; i<t; i++)
		{
			int n=Integer.parseInt(br.readLine());
			int arr[]=new int[n];
			int max=Integer.MIN_VALUE;
			long sum=0;
			
			StringTokenizer st=new StringTokenizer(br.readLine());
			
			for(int k=0; k<arr.length; k++)
				arr[k]=Integer.parseInt(st.nextToken());
			
			for(int k=arr.length-1; k>=0; k--)
			{
				if(max < arr[k])
					max=arr[k];
				if(max > arr[k])
					sum+=(max-arr[k]);
			}
			
			list.add(sum);
		}
		
		for(long num : list)
			System.out.println(num);

	}
}

좋은 웹페이지 즐겨찾기