688. Knight Probability in Chessboard
题目描述
在NxN棋盘上,骑士从第r行和第c列开始,并尝试进行K次移动。 行和列都是0索引的,所以左上角的正方形是(0,0),右下角的正方形是(N-1,N-1)。
象棋骑士有8种可能的动作,如下图所示。 每个移动都是基本方向上的两个方格,然后是正交方向上的一个方格。
骑士每移动一次,它就会随机选择八种可能的移动中的一种(即使棋子会从棋盘上移走)并在那里移动。
骑士继续前进,直到完成K次移动或已经离开棋盘。 返回骑士在停止移动后留在板上的概率。
解题思路
首先最容易想到的就是递归方法。很容易得到递归公式:
p(k, r, c) = \sum p(k-1, r+move_0, c+move_1) / 8
当k=0时有p=1,其余情况概率为8种位移的概率平均值。注意剪去不合法的移动。
但是递归的代码会超时,原因在于有很多重复计算的结果。所以要用数组储存计算加快计算。代码如下:
class Solution {
public:
double knightProbability(int N, int K, int r, int c) {
if (K == 0) {
return 1.;
}
int moves[9] = {2, 1, -2, -1, -2, 1, 2, -1, 2};
vector<vector<vector<double> > > p(K+1, vector<vector<double> >(N, vector<double>(N, 0.)));
for (int a=0; a<N; a++)
for (int b=0; b<N; b++)
p[0][a][b] = 1.;
for (int l=1; l<=K; l++)
for (int a=0; a<N; a++)
for (int b=0; b<N; b++) {
for (int i=0; i<8; i++)
if (valid(a+moves[i], b+moves[i+1], N))
p[l][a][b] += p[l-1][a+moves[i]][b+moves[i+1]];
p[l][a][b] /= 8.;
}
return p[K][r][c];
}
bool valid(int a, int b, int n) {
return a >= 0 && a < n && b >= 0 && b < n;
}
};
以前用数组的时候我总是习惯用p[100][25][25]
这种写法,但是在leetcode上容易出现内存没有清空被下一个测试样例误用导致出现错误的情况。以后要用vector
并进行初始化操作。