3008跳格子_dp

单点时限： 1.0 s

内存限制： 512 MB

题目描述

有  个点，编号为 ，在  秒时 c4Lnn 位于点 ，每秒 c4Lnn 都必须跳到其它点，即 c4Lnn 不能在同一点连续停留  秒。

求经过  秒后，c4Lnn 停留在点  的不同方案数。

由于答案会很大，请输出答案对  取模的结果。

输入格式

一行四个整数  。

输出格式

一行，一个整数，表示不同方案数。

样例

input

3 1 3 3

output

3

有以下  种方案：

因为每秒必须移动，所以在k−1秒我们不能停留在点t，容易发现我们只需判断当前是否停留在点t即可。考虑用DP解决该问题，记dpi,0/1为经过i秒后停留在点t/!t的不同方案数。

#include <bits/stdc++.h>
#define x first
#define y second
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
int c = 1e9 + 7;
ll dp[1024000][2] = {0};
// dp[i][0]是i秒时不在t点的方案数
// dp[i][1]是i秒时在t点的方案数
int main()
{
    // system("chcp 65001");
    cin.tie(0);
    cout.tie(0);
    // freopen("C:/Users/zhaochen/Desktop/input.txt", "r", stdin);
    int n, s, t, k;
    cin >> n >> s >> t >> k;
    if (s == t)
    {
        dp[0][1] = 1;
    }
    else
    {
        dp[0][0] = 1;
    }
    for (int i = 1; i <= k; i++)
    {
        dp[i][0] = (dp[i - 1][0] * (n - 2) % c + dp[i - 1][1] * (n - 1) % c) % c;
        dp[i][1] = dp[i - 1][0];
        // cout << "dp[" << i << "][0] = " << dp[i][0] << endl;
        // cout << "dp[" << i << "][1] = " << dp[i][1] << endl;
    }
    cout << dp[k][1];
    return 0;
}