LeetCode999. 可以被一步捕获的棋子数（python）

题目

在一个 8 x 8 的棋盘上，有一个白色的车（Rook），用字符 ‘R’ 表示。棋盘上还可能存在空方块，白色的象（Bishop）以及黑色的卒（pawn），分别用字符 ‘.’，‘B’ 和 ‘p’ 表示。不难看出，大写字符表示的是白棋，小写字符表示的是黑棋。
车按国际象棋中的规则移动。东，西，南，北四个基本方向任选其一，然后一直向选定的方向移动，直到满足下列四个条件之一：
~棋手选择主动停下来。
~棋子因到达棋盘的边缘而停下。
~棋子移动到某一方格来捕获位于该方格上敌方（黑色）的卒，停在该方格内。
~车不能进入/越过已经放有其他友方棋子（白色的象）的方格，停在友方棋子前。
你现在可以控制车移动一次，请你统计有多少敌方的卒处于你的捕获范围内（即，可以被一步捕获的棋子数）。

思路

R棋子最多有四种可能，上下左右都有可以直接获取的p棋子
（当然还有R棋子在四角，在边线上的情况，按上下左右都有计算其实已经包含了这些情况）
要判断R的上下左右，首先要定位R的位置

row, col = -1, -1
# 先找的'R'的位置并记录
for i in range(8):
    for j in range(8):
        if board[i][j] == 'R':
            row, col = i, j
            break

R棋子右边可以获取的p棋子

re = 0
# 右边，row保持不变，向右一格一格前进判断是否有一步直接获取的棋子p
j = col
while j < 7:
    j += 1 # j位置是R，所以先加1
    if board[row][j] == '.':
        continue # .是空格子直接跳过
    elif board[row][j] == 'B':
        break  # B不能被越过，不能再向右前进，跳出
    else:
        re += 1 # p可以直接获取 结果加1，跳出
        break

R棋子左边可以获取的p棋子

# 左边，row保持不变，向左一格一格前进判断是否有一步直接获取的棋子p
j = col
while j > 0:
    j -= 1
    if board[row][j] == '.':
        continue
    elif board[row][j] == 'B':
        break
    else:
        re += 1
        print(re)
        break

R棋子下边可以获取的p棋子

# 下边，col保持不变，向下一格一格前进判断是否有一步直接获取的棋子p
i = row
while i < 7:
    i += 1
    if board[i][col] == '.':
        continue
    elif board[i][col] == 'B':
        break
    else:
        re += 1
        break

R棋子上边可以获取的p棋子

# 上边，col保持不变，向上一格一格前进判断是否有一步直接获取的棋子p
i = row
while i > 0:
    i -= 1
    if board[i][col] == '.':
        continue
    elif board[i][col] == 'B':
        break
    else:
        re += 1
        break

题解

class Solution:
    def numRookCaptures(self, board: List[List[str]]) -> int:
        row, col = -1, -1
        # 先找的'R'的位置并记录
        for i in range(8):
            for j in range(8):
                if board[i][j] == 'R':
                    row, col = i, j
                    break
        re = 0
        # 右边，row保持不变，向右一格一格前进判断是否有一步直接获取的棋子p
        j = col
        while j < 7:
            j += 1 # j位置是R，所以先加1
            if board[row][j] == '.':
                continue # .是空格子直接跳过
            elif board[row][j] == 'B':
                break  # B不能被越过，不能再向右前进，跳出
            else:
                re += 1 # p可以直接获取 结果加1，跳出
                break
        # 左边，row保持不变，向左一格一格前进判断是否有一步直接获取的棋子p
        j = col
        while j > 0:
            j -= 1
            if board[row][j] == '.':
                continue
            elif board[row][j] == 'B':
                break
            else:
                re += 1
                print(re)
                break
        # 下边，col保持不变，向下一格一格前进判断是否有一步直接获取的棋子p
        i = row
        while i < 7:
            i += 1
            if board[i][col] == '.':
                continue
            elif board[i][col] == 'B':
                break
            else:
                re += 1
                break
        # 上边，col保持不变，向上一格一格前进判断是否有一步直接获取的棋子p
        i = row
        while i > 0:
            i -= 1
            if board[i][col] == '.':
                continue
            elif board[i][col] == 'B':
                break
            else:
                re += 1
                break
        return re

过啦~