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;
}