车的可用捕获量---------位置数组

当需要在数组中进行上下左右移动时,可以定义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

 

发布了83 篇原创文章 · 获赞 5 · 访问量 1万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 书香水墨 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览