当需要在数组中进行上下左右移动时,可以定义x、y数组用来表示移动的方向。
dx=[0,1,0,-1]
dy=[1,0,-1,0]
例题:https://leetcode-cn.com/problems/available-captures-for-rook/
在一个 8 x 8 的棋盘上,有一个白色车(rook)。也可能有空方块,白色的象(bishop)和黑色的卒(pawn)。它们分别以字符 “R”,“.”,“B” 和 “p” 给出。大写字符表示白棋,小写字符表示黑棋。
车按国际象棋中的规则移动:它选择四个基本方向中的一个(北,东,西和南),然后朝那个方向移动,直到它选择停止、到达棋盘的边缘或移动到同一方格来捕获该方格上颜色相反的卒。另外,车不能与其他友方(白色)象进入同一个方格。
返回车能够在一次移动中捕获到的卒的数量。
分析:
1、先找到白色车的下标
2、将车朝四个方向移动,直到遇到卒、白棋或者数组边缘。res记录捕获到的卒的数量。
class Solution:
def numRookCaptures(self, board: List[List[str]]) -> int:
dx=[0,1,0,-1]
dy=[1,0,-1,0]
for i in range(8):
for j in range(8):
if board[i][j]=='R':
a,b = i,j
res = 0
for k in range(4):
step = 0
while 1:
x = a+step*dx[k]
y = b+step*dy[k]
if x<0 or x>=8 or y<0 or y>=8 or board[x][y]=='B':
break
if board[x][y]=='p':
res+=1
break
step+=1
return res