백준_2579 계단 오르기(자바) / DP

시간&메모리 제한

 

문제

 

입력&출력

 

문제풀이

package com.Back;

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

public class Back_2579 {

	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 [] dp = new int[N+1];
		
		for (int i = 1; i <= N; i++) {
			arr[i]=Integer.parseInt(br.readLine());
		}
		
		if(N==1) {
			System.out.println(arr[1]);
			System.exit(0);
		}else if(N==2) {
			System.out.println(arr[1]+arr[2]);
			System.exit(0);
		}else if(N==3) {
			dp[3] = Math.max(arr[1]+arr[3], arr[2]+arr[3]);
			System.exit(0);
		}
		
		dp[1] = arr[1];
		dp[2] = arr[1] + arr[2];
		dp[3] = Math.max(arr[1]+arr[3], arr[2]+arr[3]);
		
		for (int i = 4 ; i <= N; i++) {
			dp[i] = Math.max(dp[i-3]+arr[i-1]+arr[i], dp[i-2]+arr[i]);
		}
		System.out.println(dp[N]);
	}
}

 

※ 내 생각

이 문제는 DP를 이용하는 문제입니다.

먼저 N 이 1~3 인 경우를 제외를 시켜줘야 합니다.
4 이상의 경우에는 for문을 통해서 dp를 구해줍니다.

마지막에 dp[N]을 출력해주면 됩니다.