题目
在一个 n x n 的国际象棋棋盘上,一个骑士从单元格 (row, column) 开始,并尝试进行 k 次移动。行和列是 从 0 开始 的,所以左上单元格是 (0,0) ,右下单元格是 (n - 1, n - 1) 。
象棋骑士有8种可能的走法,如下图所示。每次移动在基本方向上是两个单元格,然后在正交方向上是一个单元格。
每次骑士要移动时,它都会随机从8种可能的移动中选择一种(即使棋子会离开棋盘),然后移动到那里。
骑士继续移动,直到它走了 k 步或离开了棋盘。
返回 骑士在棋盘停止移动后仍留在棋盘上的概率 。
示例 1:
输入: n = 3, k = 2, row = 0, column = 0 输出: 0.0625 解释: 有两步(到(1,2),(2,1))可以让骑士留在棋盘上。 在每一个位置上,也有两种移动可以让骑士留在棋盘上。 骑士留在棋盘上的总概率是0.0625。
示例 2:
输入: n = 1, k = 0, row = 0, column = 0 输出: 1.00000
解题
方法一:记忆化递归
class Solution { public: vector<vector<int>> dirs = {{-2,-1},{-1,-2},{1,2},{2,1},{2,-1},{1,-2},{-1,2},{-2,1}}; vector<vector<vector<double>>> memory; double knightProbability(int n, int k, int row, int column) { memory=vector<vector<vector<double>>>(n,vector<vector<double>>(n,vector<double>(k+1))); return dfs(n,k,0,row,column); } double dfs(int n,int k,int count,int x,int y){ if(x<0||x>=n||y<0||y>=n) return 0; if(count==k) return 1; if(memory[x][y][count]) return memory[x][y][count]; double res=0; for(int i=0;i<8;i++){ int nx=x+dirs[i][0]; int ny=y+dirs[i][1]; res+=dfs(n,k,count+1,nx,ny)/8; } return memory[x][y][count]=res; } };