leetcode 688. Knight Probability in Chessboard | 688. “马”在棋盘上的概率(dp,记忆化搜索)

x33g5p2x  于2021-11-12 转载在 其他  
字(1.0k)|赞(0)|评价(0)|浏览(244)

题目

https://leetcode.com/problems/knight-probability-in-chessboard/

题解

左神讲过类似问题:
给定5个参数,N,M,row,col,k
表示在N*M的区域上,醉汉Bob初始在(row,col)位置
Bob一共要迈出k步,且每步都会等概率向上下左右四个方向走一个单位
任何时候Bob只要离开N*M的区域,就直接死亡
返回k步之后,Bob还在N*M的区域的概率

对于本题来说,k步之后留在棋盘上的概率 = 留在棋盘上的方法数/总方法数

  1. class Solution {
  2. // k步之后留在棋盘上的概率 = 留在棋盘上的方法数/总方法数
  3. public double knightProbability(int n, int k, int row, int column) {
  4. double[][][] dp = new double[n][n][k + 1]; // int overflow
  5. for (int i = 0; i < n; i++) {
  6. for (int j = 0; j < n; j++) {
  7. dp[i][j][0] = 1;
  8. for (int step = 1; step <= k; step++) {
  9. dp[i][j][step] = -1;
  10. }
  11. }
  12. }
  13. return process(n, row, column, k, dp) / Math.pow(8, k);
  14. }
  15. // 在(i,j)位置,剩余k步可以跳的情况下,返回能够留在棋盘上的方法数
  16. public double process(int n, int i, int j, int step, double[][][] dp) {
  17. if (i < 0 || i >= n || j < 0 || j >= n) return 0;
  18. if (dp[i][j][step] >= 0) return dp[i][j][step];
  19. dp[i][j][step] = process(n, i - 2, j + 1, step - 1, dp) +
  20. process(n, i - 2, j - 1, step - 1, dp) +
  21. process(n, i + 2, j + 1, step - 1, dp) +
  22. process(n, i + 2, j - 1, step - 1, dp) +
  23. process(n, i - 1, j + 2, step - 1, dp) +
  24. process(n, i - 1, j - 2, step - 1, dp) +
  25. process(n, i + 1, j - 2, step - 1, dp) +
  26. process(n, i + 1, j + 2, step - 1, dp);
  27. return dp[i][j][step];
  28. }
  29. }

相关文章