LeetCode 935. Knight Dialer

原题链接在这里：https://leetcode.com/problems/knight-dialer/

题目：

A chess knight can move as indicated in the chess diagram below:

 .           

This time, we place our chess knight on any numbered key of a phone pad (indicated above), and the knight makes N-1 hops.  Each hop must be from one key to another numbered key.

Each time it lands on a key (including the initial placement of the knight), it presses the number of that key, pressing N digits total.

How many distinct numbers can you dial in this manner?

Since the answer may be large, output the answer modulo 10^9 + 7.

Example 1:

Input: 1
Output: 10

Example 2:

Input: 2
Output: 20

Example 3:

Input: 3
Output: 46

Note:

• 1 <= N <= 5000

题解：

The question asks distinct numbers could dial.

It is actually the sum of ways jump ending at each cell.

Cell 1 could jump to cell 6 and 8. Thus accumlate the current ways count to next ways at 6 and 8.

Eventually, get all the sum.

Time Complexity: O(N).

Space: O(1).

AC Java:

 class Solution {
     public int knightDialer(int N) {
         if(N == 0){
             return 0;
         }

         if(N == 1){
             return 10;
         }

         int M = 1000000007;
         long [] cur = new long[10];
         Arrays.fill(cur, 1);
         for(int k = 2; k<=N; k++){
             long [] next = new long[10];

             next[1] = (cur[6]+cur[8])%M;
             next[2] = (cur[7]+cur[9])%M;
             next[3] = (cur[4]+cur[8])%M;
             next[4] = (cur[3]+cur[9]+cur[0])%M;
             next[5] = 0;
             next[6] = (cur[1]+cur[7]+cur[0])%M;
             next[7] = (cur[2]+cur[6])%M;
             next[8] = (cur[1]+cur[3])%M;
             next[9] = (cur[2]+cur[4])%M;
             next[0] = (cur[4]+cur[6])%M;

             cur = next;
         }

         long res = 0;
         for(int i = 0; i<10; i++){
             res = (res + cur[i]) % M;
         }
         return (int)res;
     }
 }

类似Number of Ways to Stay in the Same Place After Some Steps.