codeforce 479E Riding in a Lift

原题地址

题意

一个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]
-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]
-sum[k][b]);
}
}