题意
一个N层建筑,有一台电梯,一个人开始时位于a层,他觉得坐电梯上上下下K次,建筑的b层有一个实验室,实验室禁止进入,而且对于人每次坐电梯有限制条件
假如目前人在X层,希望坐电梯去Y层,需要满足
abs(X-Y)<abs(X-b)
问人这K次上上下下总共有多少种方案,答案对1e9+7取模
题解
DP啦,状态转移比较好搞吧,有两个问题,一个是严格大于的写法,一个是剩余系下减法要避免出现负数
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=5000;
const int H=1e9+7;
int dp[maxn+5][maxn+5];
int sum[maxn+5][maxn+5];
int main(void)
{
#ifdef ex
freopen ("in.txt","r",stdin);
#endif
int n,a,b,k;
scanf("%d%d%d%d",&n,&a,&b,&k);
dp[0][a]=1;
for (int i=a;i<=n;++i)
{
sum[0][i]=1;
}
if (a<b)
{
for (int i=1;i<=k;++i)
{
for (int j=1;j<=b-1;++j)
{
int h=ceil((b+j*1.0)/2-1);
dp[i][j]=sum[i-1][h];
if (j<=h) dp[i][j]=(dp[i][j]-dp[i-1][j]+H)%H;
sum[i][j]=(sum[i][j-1]+dp[i][j])%H;
}
}
printf("%d\n",sum[k][b-1]);
#ifdef ex1
cout<<dp[2][1]<<endl;
cout<<dp[2][2]<<endl;
cout<<dp[2][3]<<endl;
#endif
}
else
{
for (int i=1;i<=k;++i)
{
for (int j=b+1;j<=n;++j)
{
int h=floor((b+j*1.0)/2+1);
dp[i][j]=(sum[i-1][n]-sum[i-1][h-1]+H)%H;
if (j>=h) dp[i][j]=(dp[i][j]-dp[i-1][j]+H)%H;
sum[i][j]=(sum[i][j-1]+dp[i][j])%H;
}
}
printf("%d\n",sum[k][n]-sum[k][b]);
}
}