题目描述
在一个 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;
}
};