【Leetcode】935. Knight Dialer

题目地址：

https://leetcode.com/problems/knight-dialer/

给定一个如下图所示的棋盘：
允许从任何一个点出发，每次只能像国际象棋里的马一样走“日”。走$N-1$步后，将其所有踩过的数字拼接起来，就得到了一个长度为$N$的序列。问长度为$N$的不同序列有多少个。将结果模$10^9+7$后返回。

思路是动态规划。设$f[k][x]$是走$k$步恰好走到数字$x$的方案数，则问题即要求${\sum }_{x=0}^{9}f[n-1][x]$。已经知道$f[0][.]=1$，可以递推，每次将$f[i][x]$累加到满足$x\to y$的$f[i+1][y]$上。代码如下：

class Solution {
 public:
  int knightDialer(int n) {
  vector<int> v[10] = {
      {4, 6},
      {6, 8}, 
      {7, 9},
      {4, 8},
      {3, 9, 0},
      {},
      {1, 7, 0},
      {2, 6},
      {1, 3},
      {2, 4}};
    int MOD = 1e9 + 7;
    int f[2][10];
    memset(f, 0, sizeof f);
    fill(begin(f[0]), end(f[0]), 1);
    for (int i = 0; i < n - 1; i++) {
      fill(begin(f[i + 1 & 1]), end(f[i + 1 & 1]), 0);
      for (int k = 0; k <= 9; k++)
        for (int x : mp[k])
          f[i + 1 & 1][x] = (f[i + 1 & 1][x] + f[i & 1][k]) % MOD;
    }

    int res = 0;
    for (int i = 0; i <= 9; i++)
      res = (res + f[n - 1 & 1][i]) % MOD;
    return res;
  }
};

时间复杂度$O(n)$，空间$O(1)$。