LeetCode每日一题:999. 车的可用捕获量

在一个 8 x 8 的棋盘上,有一个白色车(rook)。也可能有空方块,白色的象(bishop)和黑色的卒(pawn)。它们分别以字符 “R”,“.”,“B” 和 “p” 给出。大写字符表示白棋,小写字符表示黑棋。
车按国际象棋中的规则移动:它选择四个基本方向中的一个(北,东,西和南),然后朝那个方向移动,直到它选择停止、到达棋盘的边缘或移动到同一方格来捕获该方格上颜色相反的卒。另外,车不能与其他友方(白色)象进入同一个方格。
返回车能够在一次移动中捕获到的卒的数量。
****

示例 1:
在这里插入图片描述
解题思路:车只能走上下左右,因此设置四个函数上下左右移动,遇到空格继续向对应方向移动,否则返回0
代码如下

class Solution {
public:
    int numRookCaptures(vector<vector<char>>& board) {
        int pawn_nums=0;
        int a[2];
        int hang=board.size();
        int lie=board[0].size();
        for(int i=0;i<hang;i++){
            for(int j=0;j<lie;j++){
                if(board[i][j]=='R'){//找到车的位置
                    a[0]=i;
                    a[1]=j; 
                    break;
                }
            }
        }
        pawn_nums+=upmoving(board,a);//上移吃卒
        pawn_nums+=downmoving(board,a);//下移吃卒
        pawn_nums+=rightmoving(board,a);//右移吃卒
        pawn_nums+=leftmoving(board,a);//左移吃卒
        return pawn_nums;
    }
    int upmoving(vector<vector<char>>& board,int b[2]){//上移
        int a[2];//防止改变原本车的点
        a[0]=b[0];
        a[1]=b[1];
        while(a[0]!=0){
            if(board[a[0]-1][a[1]]=='.'){
                a[0]=a[0]-1;
            }
            else if(board[a[0]-1][a[1]]=='p'){
                return 1;
            }
            else if(board[a[0]-1][a[1]]=='B'){
                return 0;
            }
        }
        return 0;
    }
    int downmoving(vector<vector<char>>& board,int b[2]){
        int a[2];
        a[0]=b[0];
        a[1]=b[1];
        while(a[0]!=board.size()-1){
            if(board[a[0]+1][a[1]]=='.'){
                a[0]=a[0]+1;
            }
            else if(board[a[0]+1][a[1]]=='p'){
                return 1;
            }
            else if(board[a[0]+1][a[1]]=='B'){
                return 0;
            }
        }
        return 0;
    }
    int leftmoving(vector<vector<char>>& board,int b[2]){
        int a[2];
        a[0]=b[0];
        a[1]=b[1];
        while(a[1]!=0){
            if(board[a[0]][a[1]-1]=='.'){
                a[1]=a[1]-1;
            }
            else if(board[a[0]][a[1]-1]=='p'){
                return 1;
            }
            else if(board[a[0]][a[1]-1]=='B'){
                return 0;
            }
        }
        return 0;
    }
    int rightmoving(vector<vector<char>>& board,int b[2]){
        int a[2];
        a[0]=b[0];
        a[1]=b[1];
        while(a[1]!=board[0].size()-1){
            if(board[a[0]][a[1]+1]=='.'){
                a[1]=a[1]+1;
            }
            else if(board[a[0]][a[1]+1]=='p'){
                return 1;
            }
            else if(board[a[0]][a[1]+1]=='B'){
                return 0;
            }
        }
        return 0;
    }
};
发布了70 篇原创文章 · 获赞 39 · 访问量 2254

猜你喜欢

转载自blog.csdn.net/weixin_45221477/article/details/105129221