LeetCode-【动态规划】-“马”在棋盘上的概率
已知一个 N
xN
的国际象棋棋盘,棋盘的行号和列号都是从0开始。即最左上角的格子记为 (0, 0)
, 最右下角的记为 (N-1, N-1)
。
现有一个“马”(也译作“骑士”)位于 (r, c)
,并打算进行 K
次移动。
如下图所示,国际象棋的“马”每一步先沿水平或垂直方向移动2个格子,然后向与之相垂直的方向再移动1个格子,共有8个可选的位置。
现在“马”每一步都从可选的位置(包括棋盘外部的)中独立随机地选择一个进行移动,直到移动了 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);
}
}