动态规划入门

会长大人非常喜欢玩《塞尔达传说:旷野之息》,有一天,手残的会长大人lj拜托游戏达人lmj帮忙开启最困难的第120个祠堂,这个祠堂的通过方法可以转化为一个一维问题。你初始在0这个位置,终点是n这个位置,你每次最多可以跳3格。很不幸的是,有些位置是有陷阱的,每踩一下扣一点血,你开始有k点血,也就是说你最多踩k-1个陷阱。问跳到终点不死有多少种跳法,保证有解。
lmj一看这个问题实在是太简单了,所以把这个任务交给了你。

动态规划入门
 

Input

单组样例
一个整数n,代表有n格(1<=n<=20)
一个整数m,代表有m个陷阱(1<=m<=n)
一个整数k,代表有k点血(1<=k<=m+1)
接下来有m个互不相同的整数,代表这些格子有一个陷阱。(1<=ai<=n)
 

Output

跳到终点有几种跳法
 

Sample Input

3 0 1

3 3 1
1 2 3

Sample Output

4

0
解题思路:
设dp[i][j]代表走到第i个位置还剩j滴血,如果当前格为陷阱,dp[i][j]=dp[i-1][j+1]+dp[i-2][j+1]+dp[i-3][j+1],
如果当前格不为陷阱,dp[i][j]=dp[i-1][j];+dp[i-2][j]+dp[i-3][j];

最后统计dp[n][1]--dp[n][k]的和即可

ac代码:


#include<bits/stdc++.h>
using namespace std;
int main()
{
    int n,m,k,i,a[25],dp[25][25],a1,j,sum=0;
    memset(a,0,sizeof(a));
    memset(dp,0,sizeof(dp));
    scanf("%d%d%d",&n,&m,&k);
    dp[0][k]=1;
    for(i=1;i<=m;i++){
        scanf("%d",&a1);
        a[a1]=1;
    }
    for(i=1;i<=n;i++){
        for(j=1;j<=k;j++){
            if(a[i]==1){
                if(i-1>=0) dp[i][j]+=dp[i-1][j+1];
                if(i-2>=0) dp[i][j]+=dp[i-2][j+1];
                if(i-3>=0) dp[i][j]+=dp[i-3][j+1];
            }
            else{
                if(i-1>=0) dp[i][j]+=dp[i-1][j];
                if(i-2>=0) dp[i][j]+=dp[i-2][j];
                if(i-3>=0) dp[i][j]+=dp[i-3][j];
            }
        }
    }
    for(i=1;i<=k;i++)
        sum+=dp[n][i];
    printf("%d",sum);
    return 0;
}