题目:
在一个 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
提示:
1 <= n <= 25
0 <= k <= 100
0 <= row, column <= n
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/knight-probability-in-chessboard
解题思路:
依次遍历所有可能的8个方向
当步数用完:
①超出范围则返回概率为 0
②在范围内则返回概率为 1
步数未用完:
①超出范围则返回概率为 0
②在范围内则返回概率为 1/8
根据以上规则判断停止移动后在棋盘里的概率,通过回溯与记忆化方法,得到结果
代码
class Solution {
public:
int steps[8][2] = {{-2, -1}, {-2, 1}, {-1, -2}, {1, -2}, {-1, 2}, {1, 2}, {2, -1}, {2, 1}};//八个移动方向
double knightProbability(int n, int k, int row, int column) {
vector<vector<vector<double>>> memory(n, vector<vector<double>>(n,vector<double>(k + 1, 0))) ;//保存在此条件下得到的概率
return noLeave(memory, n, k, row, column);//回溯
}
double noLeave(vector<vector<vector<double>>> &memory, int n, int k, int row, int column){
if(row < 0 || row >= n || column < 0 || column >= n){return 0;}//超范围概率为 0
else if(k == 0){return 1;}//未超范围步数用尽概率为 1
else if(memory[row][column][k] != 0){return memory[row][column][k];}//当前所处条件在前面已经走过并记录过则返回在此条件下得到的概率合
double result = 0;//当前所处条件没有走过则进行移动
for(auto step : steps){result += noLeave(memory, n, k - 1, row + step[0], column + step[1]) / 8;}//遍历当前位置会移动到的8个方向,求在当前条件下最终得到的在棋盘内的概率合
memory[row][column][k] = result;//记录当前条件下会得到的概率合
return result;//返回最终得到的概率总和。
}
};
作者:yi-si-cb
链接:https://leetcode-cn.com/problems/knight-probability-in-chessboard/solution/xie-shi-ji-yi-hua-sou-suo-c-by-yi-si-cb-u3g6/