Baekjoon 2579번: 계단오르기 - 런타임에러 해결

sroa.chin 2020. 4. 17. 03:12

https://www.acmicpc.net/problem/2579

2579번: 계단 오르기

계단 오르기 게임은 계단 아래 시작점부터 계단 꼭대기에 위치한 도착점까지 가는 게임이다. <그림 1>과 같이 각각의 계단에는 일정한 점수가 쓰여 있는데 계단을 밟으면 그 계단에 쓰여 있는 점수를 얻게 된다. 예를 들어 <그림 2>와 같이 시작점에서부터 첫 번째, 두 번째, 네 번째, 여섯 번째 계단을 밟아 도착점에 도달하면 총 점수는 10 + 20 + 25 + 20 = 75점이 된다. 계단 오르는 데는 다음과 같은 규칙이 있다. 계단은 한 번에 한 계단씩

www.acmicpc.net

DP 연습문제

구현하는 것은 별로 어렵지 않은데, 런타임에러가 계속 떠서 찾느라 고생했다.

계단을 연속 세 칸을 건널 수 없기 때문에 최댓값을 저장하는 배열 max의 첫번째, 두번째를 초기화 시키고 반복문이 3번째 계단부터 시작하는데, 계단이 한 칸일 경우에는 step[2]이 존재 하지 않으므로 런타임 에러가 난 것이었다.

그래서 계단의 수가 1일때는 그 계단의 값을 출력하고 강제 종료 시켜버렸다. 강제 종료 시키는 코드가 좋은 코드가 아니라.. 그냥 if else 로 처리해야 했는지도...

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

// 계단 오르기
public class Main {
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		int[] step = new int [N+1];
		long[] max = new long [N+1];
		for (int i = 0; i < N; i++) {
			step[i+1] = Integer.parseInt(br.readLine());
		}//end of input
        if(N == 1) {
			System.out.println(step[1]);
			System.exit(0);
		}
		max[1] = step[1];
		max[2] = step[1] + step[2];
		for (int i = 3; i <= N; i++) {
			max[i] = Math.max(step[i]+step[i-1]+max[i-3], step[i]+max[i-2]);
		}
		System.out.println(max[N]);
	}//end of main
}//end of class