999. 可以被一步捕获的棋子数
描述
在一个 8 x 8 的棋盘上,有一个白色的车(Rook),用字符 'R' 表示。棋盘上还可能存在空方块,白色的象(Bishop)以及黑色的卒(pawn),分别用字符 '.','B' 和 'p' 表示。不难看出,大写字符表示的是白棋,小写字符表示的是黑棋。
车按国际象棋中的规则移动。东,西,南,北四个基本方向任选其一,然后一直向选定的方向移动,直到满足下列四个条件之一:
棋手选择主动停下来。
棋子因到达棋盘的边缘而停下。
棋子移动到某一方格来捕获位于该方格上敌方(黑色)的卒,停在该方格内。
车不能进入/越过已经放有其他友方棋子(白色的象)的方格,停在友方棋子前。
你现在可以控制车移动一次,请你统计有多少敌方的卒处于你的捕获范围内(即,可以被一步捕获的棋子数)。
示例:
输入:[[".",".",".",".",".",".",".","."],[".",".",".","p",".",".",".","."],[".",".",".","p",".",".",".","."],["p","p",".","R",".","p","B","."],[".",".",".",".",".",".",".","."],[".",".",".","B",".",".",".","."],[".",".",".","p",".",".",".","."],[".",".",".",".",".",".",".","."]]
输出:3
解释:
车可以捕获位置 b5,d6 和 f5 的卒。
思路
- 方向数组
解决这种棋盘位置移动的时候都可以考虑下方向数组,上下左右xy的偏移分别为:
dx[4]={0,0,-1,1};
dy[4]={1,-1,0,0};
找到车的位置,然后使用方向数组移动,判断边界条件和是否遇到象和卒即可。
class Solution {
public:
int numRookCaptures(vector<vector<char>>& board) {
int dx[4]={0,0,-1,1};
int dy[4]={1,-1,0,0};
for(int i=0;i<board.size();i++)
for(int j=0;j<board.size();j++){
if(board[i][j]=='R'){
int count=0;
for(int k=0;k<4;k++){
int x=i,y=j;
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'){
count++;
break;
}
}
}
return count;
}
}
return 0;
}
};