백준 2579번: 계단 오르기

이 문제는 각 계단에 쓰여 있는 점수가 주어질 때 이 게임에서 얻을 수 있는 총 점수의 최댓값을 구하는 문제이다.

최댓값 구하는 문제는 DP 유형을 한번쯤 떠올려보길 바란다.

 

이 문제의 키포인트는 두가지이다.

1. n번째 계단에 도달할 수 있는 경우는 이 두가지 경우뿐이다.

             n-2번째 계단에서 2칸 점프해서 n번째 계단으로 오거나,

             n-3번째 계단에서 2칸 점프해서 n-1, n번째 계단으로 오거나.

             -> 따라서 점수는

             n-2번째 계단까지 값의 최댓값 + n번째 계단값이나

             n-3번째 계단까지 값의 최댓값 + n-1번째 계단값+ n번째 계단값 중 더 큰 값이다.

 

2. 계단의 개수가 2개 이하일 경우 base case를 만들 때 예외가 발생하므로 조건을 넣어서 처리해야한다.

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

public class Main {

	public static int sumStep(int[] step, int n) {
		int[] dp = new int[step.length+1];
		
		dp[0] = step[0];
		if(n>1) {
			dp[1] = step[0]+step[1];
		}
		if(n>2) {
			dp[2] = Math.max(step[0]+step[2],step[1]+step[2]);
		}
		
		for(int i=3; i<n; i++) {
			dp[i] = Math.max(dp[i-2]+step[i], dp[i-3]+step[i-1]+step[i]);
		}
		
		return dp[n-1];
	}
	
	public static void main(String[] args) throws IOException {
		 BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		 BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
	     StringTokenizer st = new StringTokenizer(br.readLine());
	     StringBuilder sb = new StringBuilder();
	     
	     int n = Integer.parseInt(st.nextToken());
	     
	     int[] step = new int[n];
	     
	   
	     for(int i=0; i<n; i++) {
	    	 st = new StringTokenizer(br.readLine());
	    	 step[i] = Integer.parseInt(st.nextToken());
	     }
	     
	    
	     sb.append(sumStep(step,n));
	     bw.write(sb.toString());
	     bw.flush();
	     bw.close();
	}

}