[백준][2579번] 계단 오르기

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

2579번: 계단 오르기

 

 

풀이과정

 

계단을 1칸, 2칸 전진 가능하며, 연속해서 3계단을 밟을 수 없다는 조건을 생각해보면, 아래와 같이 2가지 경우가 존재한다.

 

O는 밟은 계단, X는 건너 뛴 계단이다.

N-3	N-2	N-1	N
O,X	O	X	O
O	X	O	O

 

위의 표를 참고하여 아래와 같은 점화식을 구 할 수 있다.

v1 = dp[N-2];

v2 = dp[N-3] + map[N-1]; 

dp[N] = map[N] + max(v1, v2);

 

 

소스코드

#include <stdio.h>
#include <string.h>

int map[302];
int dp[302];

int solution(int n)
{
	if (dp[n] != -1) {
		return dp[n];
	}

	dp[n] = 0;
	if (n - 3 >= 0) {
		int tmp = solution(n - 3) + map[n - 1] + map[n];
		if (tmp > dp[n]) {
			dp[n] = tmp;
		}
	}
	if (n - 2 >= 0) {
		int tmp = solution(n - 2) + map[n];
		if (tmp > dp[n]) {
			dp[n] = tmp;
		}
	}

	return dp[n];
}
int main()
{
	int n;
	scanf("%d", &n);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &map[i]);
	}
	memset(dp, -1, sizeof(dp));
	map[0] = dp[0] = 0;
	dp[1] = map[1];
	dp[2] = map[1] + map[2];
	printf("%d\n", solution(n));
}