LeetCode 935. 骑士拨号器--动态规划

935. 骑士拨号器

国际象棋中的骑士可以按下图所示进行移动：

.

这一次，我们将 “骑士” 放在电话拨号盘的任意数字键（如上图所示）上，接下来，骑士将会跳 N-1 步。每一步必须是从一个数字键跳到另一个数字键。

每当它落在一个键上（包括骑士的初始位置），都会拨出键所对应的数字，总共按下 N 位数字。

你能用这种方式拨出多少个不同的号码？

因为答案可能很大，所以输出答案模 10^9 + 7。

示例 1：

输入：1
输出：10

示例 2：

输入：2
输出：20

示例 3：

输入：3
输出：46

提示：

1 <= N <= 5000

题解

每个位置能跑到另外一个位置，每一步累加上一步的方案数即可。

AC代码

class Solution {
public:
    typedef long long ll;
    ll mod = 1000000007;
    ll dp[10][2];
    int knightDialer(int n) 
    {
        memset(dp,0,sizeof(dp));
        for(int i=0;i<10;i++)
        dp[i][0]=1;
        for(int i=2;i<=n;i++)
        {
            dp[0][1] = dp[4][0] + dp[6][0];
            dp[1][1] = dp[6][0] + dp[8][0];
            dp[2][1] = dp[7][0] + dp[9][0];
            dp[3][1] = dp[4][0] + dp[8][0];
            dp[4][1] = dp[0][0] + dp[3][0] + dp[9][0];
            dp[6][1] = dp[0][0] + dp[1][0] + dp[7][0];
            dp[7][1] = dp[2][0] + dp[6][0];
            dp[8][1] = dp[1][0] + dp[3][0];
            dp[9][1] = dp[4][0] + dp[2][0];
            for(int j=0;j<10;j++)
            {
                dp[j][0] = dp[j][1]%mod;
            }
        }
        ll ans=0;
        for(int j=0;j<10;j++)
        {
            ans +=(dp[j][0]);
            ans %= mod;
        }
        return ans;
    }
};