力扣刷题记录#数组#简单#999车的可用捕获量

题目描述

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

示例

输入：
[[".",".",".",".",".",".",".","."],
[".",".",".",“p”,".",".",".","."],
[".",".",".",“R”,".",".",".",“p”],
[".",".",".",".",".",".",".","."],
[".",".",".",".",".",".",".","."],
[".",".",".",“p”,".",".",".","."],
[".",".",".",".",".",".",".","."],
[".",".",".",".",".",".",".","."]]
输出：3
解释：
在本例中，车能够捕获所有的卒。

输入：
[[".",".",".",".",".",".",".","."],
[".",“p”,“p”,“p”,“p”,“p”,".","."],
[".",“p”,“p”,“B”,“p”,“p”,".","."],
[".",“p”,“B”,“R”,“B”,“p”,".","."],
[".",“p”,“p”,“B”,“p”,“p”,".","."],
[".",“p”,“p”,“p”,“p”,“p”,".","."],
[".",".",".",".",".",".",".","."],
[".",".",".",".",".",".",".","."]]
输出：0
解释：
象阻止了车捕获任何卒。

输入：
[[".",".",".",".",".",".",".","."],
[".",".",".",“p”,".",".",".","."],
[".",".",".",“p”,".",".",".","."],
[“p”,“p”,".",“R”,".",“p”,“B”,"."],
[".",".",".",".",".",".",".","."],
[".",".",".",“B”,".",".",".","."],
[".",".",".",“p”,".",".",".","."],
[".",".",".",".",".",".",".","."]]
输出：3
解释：
车可以捕获位置 b5，d6 和 f5 的卒。

解答

题目描述的真的挺绕的，简单说就是车可以向上下左右四个方向走，遇到象就不能再继续走了，同一方向的车只能捕获一个。

简单方法：首先找到车的位置，然后上下左右四个方向分别搜索是否有车。对每个方向，若遇到象，则直接退回，一旦捕获到了一个车，也退回。

class Solution(object):
    def numRookCaptures(self, board):
        """
        :type board: List[List[str]]
        :rtype: int
        """
        
        # 找到车R的位置
        break_flag = False  # 记录内层循环是否break 
        for i in range(0,8):
            for j in range(0,8):
                if board[i][j] == 'R':
                    m = i
                    n = j
                    break_flag = True
                    break
            # 由于只有一个R，若内层j循环break，则外层i也break
            if break_flag:
               break
           
        count = 0
        
        # 向右搜索        
        i = n+1
        while i<8:
            if board[m][i] == 'B':
                break
            elif board[m][i] == 'p':
                count += 1
                break
            i += 1
        
        # 向左搜索
        i = n-1
        while i>0:
            if board[m][i] == 'B':
                break
            elif board[m][i] == 'p':
                count += 1
                break
            i -= 1
            
        # 向上搜索
        i = m-1
        while i>0:
            if board[i][n] == 'B':
                break
            elif board[i][n] == 'p':
                count += 1
                break
            i -= 1
            
        # 向下搜索
        i = m+1
        while i<8:
            if board[i][n] == 'B':
                break
            elif board[i][n] == 'p':
                count += 1
                break
            i += 1
            
        return count