(M)Dynamic Programming:688. Knight Probability in Chessboard

来源:互联网 发布:电脑抓阄软件 编辑:程序博客网 时间:2024/06/08 00:41


class Solution {public:    double knightProbability(int N, int K, int r, int c) {        int moves[8][2] = {{1, 2}, {1, -2}, {2, 1}, {2, -1}, {-1, 2}, {-1, -2}, {-2, 1}, {-2, -1}};        int len = N;        vector<vector<double>> dp0(len, vector<double>(len, 1));        for(int l = 0; l < K; l++) {            vector<vector<double>> dp1(len, vector<double>(len, 0));            for(int i = 0; i < len; i++) {                for(int j = 0; j < len; j++) {                    for(int *move : moves) {                        int row = i + move[0];                        int col = j + move[1];                        if(isLegal(row, col, len))                             dp1[i][j] += dp0[row][col];                    }                }            }            dp0 = dp1;        }                return dp0[r][c] / pow(8, K);     }    bool isLegal(int r, int c, int len) {        return r >= 0 && r < len && c >= 0 && c < len;    }};


阅读全文
0 0
原创粉丝点击