dp序列组和

2021.5.25

题目：n个格子，填入[1,m]的数，相邻两数的差的绝对值小于2，问有多少方案数

输入：

5
5

输出：

1161

解题思路：定义一个足够大的二维数组，a[i][j]第i个格子值为j的方案数，将第一个格子每个值的方案数都给1，从第二个格子开始计算，a[i][j]=a[i-1][j-2]+a[i-1][j-1]+a[i-1][j]+a[i-1][j+1]+a[i-1][j+2];但是要注意越界，j-2>=1，j+2<=m；最后输出最后一个格子所有值的方案数总和

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int n,m,ans=0;
    int a[105][1005]={0};
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    a[1][i]=1;
    for(int i=2;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
           for (int k = max(j - 2 , 1) ; k <= min(j + 2 , m) ; k++)
                dp[i][j] += dp[i - 1][k];
        }
    }
    for(int i=1;i<=m;i++)
        ans+=a[n][i];
    cout<<ans;
    return 0;
}