코딩테스트/다이나믹 프로그래밍(Dynamic Programming))

[Java] 백준 2670번 : 연속부분최대곱

sujin7837 2022. 5. 16. 21:07
반응형

문제

N개의 실수가 있을 때, 한 개 이상의 연속된 수들의 곱이 최대가 되는 부분을 찾아, 그 곱을 출력하는 프로그램을 작성하시오. 예를 들어 아래와 같이 8개의 양의 실수가 주어진다면,

색칠된 부분의 곱이 최대가 되며, 그 값은 1.638이다.

입력

첫째 줄은 나열된 양의 실수들의 개수 N이 주어지고, 그 다음 줄부터 N개의 수가 한 줄에 하나씩 들어 있다. N은 10,000 이하의 자연수이다. 실수는 소수점 첫째자리까지 주어지며, 0.0보다 크거나 같고, 9.9보다 작거나 같다.

출력

계산된 최댓값을 소수점 이하 넷째 자리에서 반올림하여 소수점 이하 셋째 자리까지 출력한다.

예제 입력 1

8
1.1
0.7
1.3
0.9
1.4
0.8
0.7
1.4

예제 출력 1

1.638

 

소스코드

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

public class Main {

	private static BufferedReader br;
	private static StringTokenizer st;
	
	private static int N;
	private static double []arr, dp;
	
	public static void main(String[] args) throws NumberFormatException, IOException {
		br=new BufferedReader(new InputStreamReader(System.in));
		N=Integer.parseInt(br.readLine());
		arr=new double[N];
		dp=new double[N];
		
		for(int i=0;i<N;i++) arr[i]=Double.parseDouble(br.readLine());
		dp[0]=arr[0];
		
		double result=0.0;
		for(int i=1;i<N;i++) {
			dp[i]=Math.max(arr[i], dp[i-1]*arr[i]);
			if(dp[i]>result) result=dp[i];
		}
		
		System.out.printf("%.3f", result);
	}

}

dp[0]=arr[0]

dp[1]=max(dp[0]*arr[1], arr[1]);

...

dp[n]=max(dp[n-1]*arr[n], arr[n]);

 

곱이 최대가 되려면 이전 수와 곱한 값과 곱하지 않은 값을 비교했을 때 더 큰 값을 선택해야 합니다. 따라서 위와 같은 식을 세워서 dp에 저장하고, 전체 dp값들 중 최댓값을 구해주었습니다.

반응형