JAVA程序设计：骑士拨号器（LeetCode：935）

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

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

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

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

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

 

示例 1：

输入：1
输出：10
示例 2：

输入：2
输出：20
示例 3：

输入：3
输出：46
 

提示：

1 <= N <= 5000

思路：dp[i][j]表示第i步走到j时的方案数。

class Solution {
    public int knightDialer(int N) {
    	
    	int mod=1000000007;
    	int[][] map=new int[][] {
    		{4,6},{6,8},{7,9},{4,8},{3,9,0},
    		{},{1,7,0},{2,6},{1,3},{2,4}
    	};
    	
    	int[][] dp=new int[2][10];
    	Arrays.fill(dp[0], 1);
    	for(int i=0;i<N-1;i++) {
    		Arrays.fill(dp[~i&1], 0);
    		for(int j=0;j<10;j++)
    			for(int nei : map[j]) {
    				dp[~i&1][nei]+=dp[i&1][j];
    				dp[~i&1][nei]%=mod;
    			}
    	}
    	
    	long ans=0;
    	for(int x : dp[~N&1])
    		ans+=x;
    	return (int)(ans%mod);
    }
}