Knight Probability in Chessboard

题目描述

解题思路

题目大意如下：一个N*N的棋盘，给定一个骑士的初始位置(r,c)，再给一个步数限制K，表示这个骑士刚好可以走K步（每一步有8种走法，即“日”字型走法）,问走过K步后还在棋盘内的可能。
这道题目和out of boundary path有异曲同工之处，因为每一步都有8种走法，那么K步就有8^K种可能，而我们只需要找出在这8^K种可能里有多少种是还未走出边界的，解法类似，但这里为了节省空间，少掉一个维度，即将步数视为一个时间维度，每走一次，dp[i][j]就更新一次，而每次更新之前，都需一个temp[i][j]来记录走这一次之后的没走出边界的情况总和，循环棋盘上的每个位置之后再赋值给dp[i][j]。

C++代码实现

class Solution {
public:
    double knightProbability(int N, int K, int r, int c) {
        if (K == 0) return 1;
        vector<vector<double> > dp(N, vector<double>(N, 1));
        int move[8][2] = {{-1,-2},{-2,-1},{-2,1},{-1,2},{1,2},{2,1},{2,-1},{1,-2}};
        for (int k = 0; k < K; ++k) {
            vector<vector<double> > t(N, vector<double>(N, 0));
            for (int i = 0; i < N; ++i) {
                for (int j = 0; j < N; ++j) {
                    for (int m = 0; m < 8; ++m) {
                        int x = i + move[m][0], y = j + move[m][1];
                        if (x >= 0 && x <= N-1 && y >= 0 && y <= N-1)
                            t[i][j] += dp[x][y];
                    }
                }
            }
            dp = t;
        }
        return dp[r][c] / pow(8, K);
    }
};