480 C. Riding in a Lift

蛮简单的dp.

#include <cstdio>

const int MOD = 1000000007;

void add(int &x, int y)
{
x += y;
while (x < 0) {
x += MOD;
}
while (x >= MOD) {
x -= MOD;
}
}

int dp[5005][5005];
int n, a, b, k;

void update(int i, int a, int b, int v)
{
if (a > b)  return;
if (a < 0)  a = 1;
if (b > n)  b = n;
add(dp[i][a], v);
add(dp[i][b + 1], -v);
}

void update_again(int i)
{
for (int j = 1; j <= n; ++ j) {
add(dp[i][j], dp[i][j - 1]);
}
}

int main()
{
scanf("%d%d%d%d", &n, &a, &b, &k);

dp[0][a] = 1;

for (int i = 0; i < k; ++ i) {
if (i)  update_again(i);
for (int j = 1; j <= n; ++ j) {
if (dp[i][j] != 0) {
if (j < b) {
update(i + 1, 2 * j - b + 1, j - 1, dp[i][j]);
update(i + 1, j + 1, b - 1, dp[i][j]);
}
if (j > b) {
update(i + 1, b + 1, j - 1, dp[i][j]);
update(i + 1, j + 1, 2 * j - b - 1, dp[i][j]);
}
}
}
}
update_again(k);
int ans = 0;
for (int i = 1; i <= n; ++ i) {
add(ans, dp[k][i]);
}
printf("%d\n", ans);
return 0;
}