leetcode 999. Available Captures for Rook 车的可用捕获量

leetcode 2020年3月 每日一题打卡
无营养

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/available-captures-for-rook

代码：

时间复杂度O(n ^2)+O(n)=O(n ^2)

class Solution(object):
    def numRookCaptures(self, board):
        """
        :type board: List[List[str]]
        :rtype: int
        """

        # 先找到R的位置
        x=0
        y=0 #R的坐标
        flag=False
        for i in range(8):
            if flag:
                break
            for j in range(8):
                if board[i][j]=="R":
                    x=i
                    y=j
                    flag=True
                    break
        #判断R的上下左右
        s=0
        for k in range(1,x+1):
            if board[x-k][y]=="p":
                s+=1
                break
            elif board[x-k][y]=="B":
                break
        for k in range(1,y+1):
            if board[x][y-k]=="p":
                s+=1
                break
            elif board[x][y-k]=="B":
                break
        for k in range(1,8-x):
            if board[x+k][y]=="p":
                s+=1
                break
            elif board[x+k][y]=="B":
                break
        for k in range(1,8-y):
            if board[x][y+k]=="p":
                s+=1
                break
            elif board[x][y+k]=="B":
                break
        
        return s

本博客为原创作品，欢迎指导,转载请说明出处，附上本文链接，谢谢！