leetcode 688. “马”在棋盘上的概率【三维的动态规划】

最新推荐文章于 2025-04-01 21:52:54 发布

原创最新推荐文章于 2025-04-01 21:52:54 发布 · 293 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

103 篇文章

订阅专栏

中等难度

53 篇文章

订阅专栏

本文介绍了一种使用三维动态规划解决国际象棋棋盘上骑士移动问题的方法。通过递归地计算每一步下，单个棋格的出界概率，最终得到骑士在指定步数内留在棋盘上的概率。

执行用时 : 16 ms, 在Knight Probability in Chessboard的C++提交中击败了85.98% 的用户

内存消耗 : 10.5 MB, 在Knight Probability in Chessboard的C++提交中击败了76.67% 的用户

最开始想用两维度动态规划，把每个坐标的移动后位于棋盘内的“马”数量储存起来，奈何爆int了。（8^30，不爆才怪）

既然不能往大的走，就只能往小的走——存每一步下，单个棋格的情况：

如下：【三维动态规划】

class Solution {
public:
    //八个方向
    vector<pair<int, int>> directions = {{1, 2}, {-1, 2}, {1, -2}, {-1, -2}, {2, 1}, {-2, 1}, {2, -1}, {-2, -1}};
    double knightProbability(int N, int K, int r, int c) {
        //board[row][col][j]用于存储在row行，col列走j步不出界的概率
        vector<vector<vector<double>>> board(N, vector<vector<double>>(N, vector<double>(K + 1, 0)));
        return nextStep(board, N, K, r, c, K);
    }
    //step代表的是还需要走的步数，[nowRow][nowCol]是现在的位置
    double nextStep(vector<vector<vector<double>>> &board, int N, int K, int nowRow, int nowCol, int step){
        if (nowRow < 0 || nowCol < 0 || nowRow >= N || nowCol >= N){//出界
            return 0;
        }
        //如果此时step == 0，代表它已经不能移动，所以没有出界的概率就是1.0
        if (step == 0){
            return 1.0;
        }
        //如果这个点之前搜索过，（注意double类型不能直接判断board[nowRow][nowCol][step] == 0，需要判断它是否小于一个很小的值，比如1e-6）
        if (abs(board[nowRow][nowCol][step]) > 1e-6){
            return board[nowRow][nowCol][step];
        }
        //穷举八个方向
        for (auto &direction : directions){
            board[nowRow][nowCol][step] += nextStep(board, N, K, nowRow + direction.first, nowCol + direction.second, step - 1) / 8;
        }
        return board[nowRow][nowCol][step];
    }
};