leetcode688. “马”在棋盘上的概率

tech2024-03-28  16

传送门

题目:已知一个 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。

令 dp[k][i][j]代表马在位置 (i, j) 移动了k次以后还留在棋盘上的概率, 和576做法一模一样,同样也可以把K去掉优化dp数组变成二维的,这里省略了。

public double knightProbability(int N, int K, int r, int c) { double[][][] dp = new double[K+1][N][N]; // 注意是double,存概率 int[][] dir = new int[][]{{-2, -1},{-2,1},{2, -1},{2, 1}, {-1, -2}, {1, -2}, {-1, 2}, {1, 2}}; dp[0][r][c] = 1.0; // dp初始化 for (int k = 1; k <= K; ++k) { for (int i = 0; i < N; ++i) { for (int j = 0; j < N; ++j) { for (int d = 0; d < 8; ++d) { int newX = i + dir[d][0]; int newY = j + dir[d][1]; if (newX < 0 || newX >= N || newY < 0 || newY >= N) continue; dp[k][i][j] += dp[k - 1][newX][newY] / 8.0; } } } } double s = 0; for (int i = 0; i < N; ++i) { for (int j = 0; j < N; ++j){ s += dp[K][i][j]; // K是固定的,输入值 } } return s; }
最新回复(0)