codeforces 479e Riding in a Lift | dp

题意：

一个人乘电梯，一开始他在a层，其中b层为不能进入。现在他要进行k次冒险。要求：|x - y| < | x - b| （x为当前所在楼层，y为准备前往的楼层）。问k次旅行他的方案数是多少。

思路：

一开始想了下n^3的复杂度。。。后来看了别人的思路，开个数组，保存前缀和即可。

sum[j]：到达前 j 层的总的方案数。每次都会更新一遍。

dp[i][j] : 第 i 次旅行所在楼层为 j 的方案数。

转移到dp[i][j]:

如果 j < b，加上 ( j+b) / 2 层到一层的sum和。

如果j > b，加上顶层到（j+b）/2层的sum和。

细节自己处理一下。

code：

/* **********************************************
Created Time: 2014-10-20 18:16:57
File Name   : cf.cpp
*********************************************** */
//#pragma comment(linker, "/STACK:102400000,102400000")
#include <iostream>
#include <fstream>
#include <cstring>
#include <climits>
#include <deque>
#include <cmath>
#include <queue>
#include <stack>
#include <list>
#include <map>
#include <set>
#include <utility>
#include <sstream>
#include <complex>
#include <string>
#include <vector>
#include <cstdio>
#include <bitset>
#include <functional>
#include <algorithm>
using namespace std;
typedef long long LL;
const int MOD = 1e9+7;
const int MAXN = 5005;
int n, a, b, k;
LL dp[MAXN][MAXN];
LL sum[MAXN];
int main()
{
        cin>>n>>a>>b>>k;
        //memset(dp, 0, sizeof(dp));
        //
        dp[0][a] = 1;
        for(int i = 1;i <= k ; i++)
        {
                //
                //sum[0] = 0;
                for(int j = 1;j <= n; j++)
                        sum[j] = (dp[i-1][j]+sum[j-1]+MOD)%MOD;
                if(a < b)
                {
                        for(int j = 1;j <= b-1; j++)
                        {
                                int tt;
                                if((j+b)%2 == 0)
                                        tt = (j+b)/2 - 1;
                                else
                                        tt = floor(1.0*(j+b)/2);
                                //if(tt == b-1) tt = jifnw;
                                dp[i][j] = (dp[i][j] + (sum[tt] - dp[i-1][j])%MOD + MOD)%MOD;
                        }
                }
                else
                {
                        for(int j = b+1;j <= n; j++)
                        {
                                int tt;
                                if((j+b)%2 == 0)
                                        tt = (j+b)/2 + 1;
                                else
                                        tt = ceil(1.0*(j+b)/2);
                                dp[i][j] = (dp[i][j] + (sum[n] - sum[tt-1] - dp[i-1][j]) % MOD+ MOD)%MOD;
                        }
                }
        }
        LL res = 0;
        for(int i = 1;i <= n; i++)
                res = (res+dp[k][i])%MOD;
        cout<<res<<endl;
        return 0;
}