力扣 688. 骑士在棋盘上的概率

题目来源：https://leetcode-cn.com/problems/knight-probability-in-chessboard/

大致题意：
给一个 n*n 的矩阵棋盘和棋子的起始位置，棋子按照国际象棋规则移动，求出 k 次移动后仍在棋盘上的概率

思路

反向思考，起始位置 k 次移动后仍在棋盘上的概率可以转化为从棋盘上任意一个位置移动 k 次到达起始位置的概率

于是可以使用动态规划来解题

动态规划

1. 使用 dp[s][i][j] 表示第 s 步移动到位置 (i, j) 的概率
2. 初始时，即 s 为 0 时，所有的概率为 1
3. 更新时， dp[s][i][j] 为所有 dp[s - 1][x][y] / 8 的和，其中 (x, y) 即为可以移动到 (i, j) 的位置

代码：

public double knightProbability(int n, int k, int row, int column) {
        // 方向
        int[][] directions = {{-2, -1}, {-2, 1}, {2, -1}, {2, 1}, {-1, -2}, {-1, 2}, {1, -2}, {1, 2}};
        double[][][] dp = new double[k + 1][n][n];
        for (int s = 0; s <= k; s++) {
            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    // 初始化
                    if (s == 0) {
                        dp[s][i][j] = 1;
                        continue;
                    }
                    // 更新
                    for (int l = 0; l < 8; l++) {
                        int x = i + directions[l][0];
                        int y = j + directions[l][1];
                        if (x >= 0 && x < n && y >= 0 && y < n) {
                            dp[s][i][j] += dp[s - 1][x][y] / 8;
                        }
                    }
                }
            }
        }
        return dp[k][row][column];
    }