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

解题思路

逆向思考：dp[i][j][k]表示从某个点走k步回到(i,j)的概率。
那么一开始dp[i][j][0]=1.0（其中(i,j)在盘内），若(i,j)不在盘内，则dp[i][j][0]=0.0;
dp[i][j][k]=(8个方向的k-1之和)/8.0 。除以8.0是因为，有8种选择可以回到(i,j)，走任何一条路径的概率是1/8

代码

class Solution {
public:
    double knightProbability(int N, int K, int r, int c) {
        double dp[N+5][N+5][K+1];
        memset(dp,0,sizeof(dp));
        for(int i=0;i<N+4;i++)
        {
            for(int j=0;j<N+4;j++)
            {
                if(i>=2&&i<=N+1&&j>=2&&j<=N+1)
                {
                    dp[i][j][0]=1.f;
                }
                else
                {
                    dp[i][j][0]=0.f;
                }
            }
        }
        for(int k=1;k<=K;k++)
        {
            for(int i=2;i<=N+1;i++)
            {
                for(int j=2;j<=N+1;j++)
                {
                    dp[i][j][k]=(dp[i-2][j+1][k-1]+dp[i-2][j-1][k-1]+dp[i+2][j+1][k-1]+dp[i+2][j-1][k-1]+dp[i+1][j-2][k-1]+dp[i-1][j-2][k-1]+dp[i+1][j+2][k-1]+dp[i-1][j+2][k-1])/8.0;
                }
            }
        }
        return dp[r+2][c+2][K];
    }
};