Leetcode 688. “马”在棋盘上的概率 C++

Leetcode 688. “马”在棋盘上的概率

题目

已知一个 NxN 的国际象棋棋盘，棋盘的行号和列号都是从 0 开始。即最左上角的格子记为 (0, 0)，最右下角的记为 (N-1, N-1)。

现有一个 “马”（也译作 “骑士”）位于 (r, c) ，并打算进行 K 次移动。

如下图所示，国际象棋的 “马” 每一步先沿水平或垂直方向移动 2 个格子，然后向与之相垂直的方向再移动 1 个格子，共有 8 个可选的位置。

现在 “马” 每一步都从可选的位置（包括棋盘外部的）中独立随机地选择一个进行移动，直到移动了 K 次或跳到了棋盘外面。

求移动结束后，“马” 仍留在棋盘上的概率。

示例：

输入: 3, 2, 0, 0
输出: 0.0625
解释: 
输入的数据依次为 N, K, r, c
第 1 步时，有且只有 2 种走法令 “马” 可以留在棋盘上（跳到（1,2）或（2,1））。对于以上的两种情况，各自在第2步均有且只有2种走法令 “马” 仍然留在棋盘上。
所以 “马” 在结束后仍在棋盘上的概率为 0.0625。

注意：

• N 的取值范围为 [1, 25]
• K 的取值范围为 [0, 100]
• 开始时，“马” 总是位于棋盘上

题解

动态规划
dp[i][x][y]表示第i步马在x、y位置的概率
根据马走日的特点，我们可以知道马走一步有8种可能，分别为{-2,-1},{-2,1},{-1,2},{1,2},{2,1},{2,-1},{1,-2},{-1,-2}。因此，走1种的可能性为1/8。故转移方程就是

if(i+dir[k][0]>=0 && i+dir[k][0]<N && j+dir[k][1]>=0 && j+dir[k][1]<N)
                            dp2[i][j] += dp[i+dir[k][0]][j+dir[k][1]]/8;

其中dp2表示下一步的的一个概率信息，dp是当前步的一个概率信息。
详细过程见代码

代码

	double knightProbability(int N, int K, int r, int c) {
        vector<vector<double>> dp(N,vector<double>(N,0));
        vector<vector<double>> dp2;
        dp[r][c] = 1;
        int dir[8][2] = {
            {-2,-1},{-2,1},{-1,2},{1,2},{2,1},{2,-1},{1,-2},{-1,-2}
        };
        while(K--){
            dp2 =  vector<vector<double>>(N,vector<double>(N,0));
            for(int i=0; i<N; i++){
                for(int j=0; j<N; j++){
                    for(int k=0; k<8; k++){
                        if(i+dir[k][0]>=0 && i+dir[k][0]<N && j+dir[k][1]>=0 && j+dir[k][1]<N)
                            dp2[i][j] += dp[i+dir[k][0]][j+dir[k][1]]/8;
                    }
                }
            }
            dp = dp2;
        }
        double ans;
        for(int i=0; i<N; i++){
            for(int j=0; j<N; j++)
                ans += dp[i][j];
        }
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/knight-probability-in-chessboard
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。