[LeetCode] 999、車的可用捕獲量

題目描述

在一個 8 x 8 的棋盤上，有一個白色車（rook）。也可能有空方塊，白色的象（bishop）和黑色的卒（pawn）。它們分別以字符 “R”，“.”，“B” 和 “p” 給出。大寫字符表示白棋，小寫字符表示黑棋。車按國際象棋中的規則移動：它選擇四個基本方向中的一個（北，東，西和南），然後朝那個方向移動，直到它選擇停止、到達棋盤的邊緣或移動到同一方格來捕獲該方格上顏色相反的卒。另外，車不能與其他友方（白色）象進入同一個方格。返回車能夠在一次移動中捕獲到的卒的數量。

輸入：[[".",".",".",".",".",".",".","."],
[".",".",".","p",".",".",".","."],
[".",".",".","p",".",".",".","."],
["p","p",".","R",".","p","B","."],
[".",".",".",".",".",".",".","."],
[".",".",".","B",".",".",".","."],
[".",".",".","p",".",".",".","."],
[".",".",".",".",".",".",".","."]]
輸出：3
解釋： 
車可以捕獲位置 b5，d6 和 f5 的卒。

解題思路

模擬題，意即：將題目的意思直接實現出來，沒有使用到特定的數據結構和方法。這些問題常常作爲競賽問題的第一個問題，也叫「簽到題」。

參考代碼

class Solution {
public:
    int numRookCaptures(vector<vector<char>>& board) {
        // 定義上下左右四個方向
        int dx[] = {-1, 1, 0, 0};
        int dy[] = {0, 0, -1, 1};
       
        for (int i = 0; i < 8; i++) {
            for (int j = 0; j < 8; j++) {
                // 找到白車所在的位置
                if (board[i][j] == 'R') {
                    int res = 0;
                    // 分別判斷白車的上、下、左、右四個方向。
                    for (int k = 0; k < 4; k++) {
                        int x = i, y = j;
                        while (true) {  // 這裏的while(true)是關鍵
                            x += dx[k];
                            y += dy[k];
                            // 出界或遇到白色的象（隊友），則不再往前走。
                            if (x < 0 || x >= 8 || y < 0 || y >= 8 || board[x][y] == 'B') {
                                break;
                            }
                            // 遇到黑色的卒，計數加一，停止往前走。
                            if (board[x][y] == 'p') {
                                res++;
                                break;
                            }
                        }
                    }
                    return res;
                }
            }
        }
        
        return 0;
    }
};