leetcode 935 Knight Dialer

1.题目描述

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

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

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

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

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

示例 1：

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

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

输入：3
输出：46

范围：

1 <= N <= 5000

2.解题思路

想到DP，每多一步的电话数字与上一步的电话数字相关。
用phone来存储每个数字下一步可能走上的数字，res表示某一步数将走上其索引所在数字的所有情况的总数，new用来更新其总数情况。

3.Python代码

class Solution:
    def knightDialer(self, N):
        """
        :type N: int
        :rtype: int
        """
        if N == 1: return 10
        
        mod = 10**9 + 7
        
        phone = [[]] * 10
        phone[0] = [4,6]
        phone[1] = [6,8]
        phone[2] = [7,9]
        phone[3] = [4,8]
        phone[4] = [3,9,0]
        phone[5] = []
        phone[6] = [1,7,0]
        phone[7] = [2,6]
        phone[8] = [1,3]
        phone[9] = [2,4]

        res = [1] * 10

        for i in range(N-1):
            new = [0] * 10
            for number in range(10):
                for j in phone[number]:
                    new[j] += res[number]
            res = new

        return sum(res) % mod