leetcode 999. 车的可用捕获量

最新推荐文章于 2020-03-26 22:17:57 发布

hhmy77

最新推荐文章于 2020-03-26 22:17:57 发布

阅读量203

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/hhmy77/article/details/105124365

版权

leetcode 专栏收录该内容

110 篇文章 0 订阅

订阅专栏

在这里插入图片描述

本来以为要BFS，读完题后发现不用那么复杂
直接无脑四个循环

class Solution:
    def numRookCaptures(self, board: List[List[str]]) -> int:
        ans=0        
        for i in range(8):
            for j in range(8):
                # 计算四个方向的棋子的距离，如果先碰到象则跳过
                if board[i][j]=='R':
                    # 往左走
                    for tj in range(j-1,-1,-1):
                        if board[i][tj]=='B':
                            break
                        if board[i][tj]=='p':
                            ans+=1
                            break
                    for ti in range(i-1,-1,-1):
                        if board[ti][j]=='B':
                            break
                        if board[ti][j]=='p':
                            ans+=1
                            break
                    for tj in range(j+1,8):
                        if board[i][tj]=='B':
                            break
                        if board[i][tj]=='p':
                            ans+=1
                            break
                    for ti in range(i+1,8):
                        if board[ti][j]=='B':
                            break
                        if board[ti][j]=='p':
                            ans+=1
                            break
        return ans

在这里插入图片描述
优化下

class Solution:
    def numRookCaptures(self, board: List[List[str]]) -> int:
        ans=0        
        d=[[0,-1],[-1,0],[0,1],[1,0]]
        for i in range(8):
            for j in range(8):
                # 计算四个方向的棋子的距离，如果先碰到象则跳过
                if board[i][j]=='R':
                    for t in range(4):
                        td=d[t]
                        x,y=i,j
                        while True:
                            x+=td[0]
                            y+=td[1]
                            if x<0 or y<0 or x>=8 or y>=8 or board[x][y]=='B':
                                break
                            if board[x][y]=='p':
                                ans+=1
                                break
        return ans