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
本博客为原创作品,欢迎指导,转载请说明出处,附上本文链接,谢谢!