LeetCode - 999. 车的可用捕获量

999. 车的可用捕获量

在一个 8 x 8 的棋盘上，有一个白色车（rook）。也可能有空方块，白色的象（bishop）和黑色的卒（pawn）。它们分别以字符 “R”，“.”，“B” 和 “p” 给出。大写字符表示白棋，小写字符表示黑棋。

车按国际象棋中的规则移动：它选择四个基本方向中的一个（北，东，西和南），然后朝那个方向移动，直到它选择停止、到达棋盘的边缘或移动到同一方格来捕获该方格上颜色相反的卒。另外，车不能与其他友方（白色）象进入同一个方格。

返回车能够在一次移动中捕获到的卒的数量。

解题思路： 此题不难，按照题意，依次扫描四个方向即可。

class Solution {
public:
    int numRookCaptures(vector<vector<char>>& board) {
        int res = 0;
        for (int i = 0; i < 8; ++i) {
            for (int j = 0; j < 8; ++j) {
                if (board[i][j] == 'R') {
                    int r = i, c = j;
                    while (r >=0 && board[r][c] != 'B') {
                        if (board[r][c] == 'p') {
                            ++res;
                            break;
                        }
                        --r;
                    }
                    r = i;
                    while (r < 8 && board[r][c] != 'B') {
                        if (board[r][c] == 'p') {
                            ++res;
                            break;
                        }
                        ++r;
                    }
                    r = i;
                    while (c >= 0 && board[r][c] != 'B') {
                        if (board[r][c] == 'p') {
                            ++res;
                            break;
                        }
                        --c;
                    }
                    c = j;
                    while (c < 8 && board[r][c] != 'B') {
                        if (board[r][c] == 'p') {
                            ++res;
                            break;
                        }
                        ++c;
                    }
                    return res;
                }
            }
        }
        return 0;
    }
};