1000-骑士在棋盘上的概率

林林林ZEYU

于 2022-02-26 21:14:06 发布

阅读量147

点赞数 1

分类专栏：个人刷题文章标签：动态规划算法

本文链接：https://blog.csdn.net/LINZEYU666/article/details/123156170

版权

个人刷题专栏收录该内容

172 篇文章 8 订阅

订阅专栏

该博客讨论了一个关于国际象棋骑士在nxn棋盘上移动的问题。骑士从(row, column)位置开始，随机进行k次移动。每步移动遵循骑士的8种可能路径。问题求解的是骑士在完成k步移动后仍留在棋盘上的概率。通过动态规划的方法，建立一个三维数组dp来存储每步移动后骑士在每个位置的概率，最终返回dp[k][row][column]作为答案。复杂度分析未给出。

摘要由CSDN通过智能技术生成

题目如下

在一个 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

解题思路

动态规划
在这里插入图片描述

class Solution 
{
public:
    vector<vector<int>> dirs = { {-2, -1}, {-2, 1}, {2, -1}, {2, 1}, {-1, -2}, {-1, 2}, {1, -2}, {1, 2} };

    double knightProbability(int n, int k, int row, int column) 
    {
        vector<vector<vector<double>>> dp(k + 1, vector<vector<double>>(n, vector<double>(n)));
        for (int step = 0; step <= k; step++)
        {
            for (int i = 0; i < n; i++)
            {
                for (int j = 0; j < n; j++) 
                {
                    if (step == 0)
                    {
                        dp[step][i][j] = 1;
                    }
                    else 
                    {
                        for (auto& dir : dirs) 
                        {
                            int ni = i + dir[0], nj = j + dir[1];
                            if (ni >= 0 && ni < n && nj >= 0 && nj < n) 
                            {
                                dp[step][i][j] += dp[step - 1][ni][nj] / 8;
                            }
                        }
                    }
                }
            }
        }
        return dp[k][row][column];
    }
};