백준_2616 소형기관차(자바) / 누적합

시간&메모리 제한

 

문제

 

입력&출력

 

 

문제풀이

package com.Back;

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

public class Back_2616 {

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		int arr[] = new int[N+1];
		int sum[] = new int[N+1];
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		for (int i = 1; i <= N; i++) {
			arr[i]=Integer.parseInt(st.nextToken());
			sum[i]=sum[i-1]+arr[i];
		}
		
		int cnt = Integer.parseInt(br.readLine());
		int dp[][] = new int[4][N+1];
		
		for (int i = 1; i < 4; i++) {
			for (int j = i*cnt; j <=N; j++) {
				dp[i][j] = Math.max(dp[i][j-1], dp[i-1][j-cnt]+(sum[j]-sum[j-cnt]));
			}
		}
		System.out.println(dp[3][N]);
	}
}

 

※ 내 생각

이 문제는 누적합과 DP를 같이 이용하는 문제입니다.

먼저 arr에 각각의 손님의 수를 담아줍니다.
셋째줄에 입력 값은 전체 3으로 나눈것 보다 작기 때문에 중복되는 걱정을 할 필요가 없습니다.

sum에 arr에 대한 누적값을 담아줍니다.
DP를 이용해서 3개의 소형 기관차에 들어갈 수 있는 손님의 수를 넣어줍니다.

마지막으로 출력하면 됩니다.