Leetcode之 车的可用捕获量

题目描述

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

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

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

思路

首先找到车的位置，之后让车从起点出发往东南西北出发，碰壁或碰到白象就返回0，碰到黑卒就返回1结束。双百通过

代码

class Solution {
public:
    int numRookCaptures(vector<vector<char>>& board) {
        if(board.size()==0)
            return 0;
        int y_max = board.size()-1;
        int x_max = board[0].size()-1;
        int i = 0, j = 0;
        int ans = 0;
        for(; i < x_max;i++)
        {
            for(;j <y_max;j++)
            {
                if(board[i][j] == 'R')
                {
                    ans += move(i,j,x_max,y_max,board,1);
                    ans += move(i,j,x_max,y_max,board,2);
                    ans += move(i,j,x_max,y_max,board,3);
                    ans += move(i,j,x_max,y_max,board,4);
                    break;
                }
            }
            j = 0;
        }
        return ans;
    }
    int move(int i, int j, int x_max, int y_max, vector<vector<char>>& board, int dir)
    {
        if(i >=0 && i<= x_max && j>=0 && j<=y_max && board[i][j]!='B')
        {
            if(board[i][j] == 'p')
                return 1;
            else
            {
                if(dir == 1)
                    return move(i-1,j,x_max,y_max,board,dir);
                else if(dir == 2)
                    return move(i+1,j,x_max,y_max,board,dir);
                else if(dir == 3)
                    return move(i,j-1,x_max,y_max,board,dir);
                else if(dir == 4)
                    return move(i,j+1,x_max,y_max,board,dir);
                else
                    return 0;
            }
        }
        else
            return 0;
    }
};