LeetCode-【动态规划】-“马”在棋盘上的概率

已知一个 NxN 的国际象棋棋盘,棋盘的行号和列号都是从0开始。即最左上角的格子记为 (0, 0), 最右下角的记为 (N-1, N-1)。 

现有一个“马”(也译作“骑士”)位于 (r, c) ,并打算进行 K 次移动。 

如下图所示,国际象棋的“马”每一步先沿水平或垂直方向移动2个格子,然后向与之相垂直的方向再移动1个格子,共有8个可选的位置。

LeetCode-【动态规划】-“马”在棋盘上的概率

现在“马”每一步都从可选的位置(包括棋盘外部的)中独立随机地选择一个进行移动,直到移动了 K 次或跳到了棋盘外面。

求移动结束后,“马”仍留在棋盘上的概率。

例:

输入: 3, 2, 0, 0
输出: 0.0625
解释: 
输入的数据依次为 N, K, r, c
第1步时,有且只有2种走法令“马”可以留在棋盘上(跳到(1,2)或(2,1))。对于以上的两种情况,各自在第2步均有且只有2种走法令“马”仍然留在棋盘上。
所以“马"在结束后仍在棋盘上的概率为0.0625。

注意:

  • N 的取值范围为 [1, 25]
  • K 的取值范围为 [0, 100]
  • 开始时,“马”总是位于棋盘上

题解:按照正常思路来,对马走的每一步进行遍历,记录那些在棋盘中的点,最后再进行计算。这个解答和很详细,参考解答:https://blog.****.net/shanshanhi/article/details/79979250

class Solution {
    public double knightProbability(int N, int K, int r, int c) {
        double[][] dp=new double[N][N];
        for(double[] d:dp){
            Arrays.fill(d,1);
        }
        int[][] dt={{-1,-2},{-2,-1},{-2,1},{-1,2},{1,2},{2,1},{2,-1},{1,-2}};
        for(int k=1;k<=K;k++){
            double[][] t=new double[N][N];
            for(int i=0;i<N;i++){
                for(int j=0;j<N;j++){
                    for(int[] d:dt){
                        int x=i+d[0];
                        int y=j+d[1];
                        if(x<0||x>=N||y<0||y>=N)
                            continue;
                        t[i][j]+=dp[x][y];
                    }
                }
            }
            dp=t;
        }
        return dp[r][c]/Math.pow(8,K);
    }
}