999. 可以被一步捕获的棋子数（简单）- LeetCode

最新推荐文章于 2024-12-06 09:58:37 发布

不学无术的小镇做题家

最新推荐文章于 2024-12-06 09:58:37 发布

阅读量177

点赞数

分类专栏： LeetCode 文章标签：数据结构 python leetcode 算法

本文链接：https://blog.youkuaiyun.com/Karty9/article/details/107022582

版权

LeetCode 专栏收录该内容

52 篇文章

订阅专栏

题目描述

在这里插入图片描述

自己解法

先找到车的位置，再遍历四个方向即可。

class Solution:
    def numRookCaptures(self, board: List[List[str]]) -> int:
        ans = 0
        x,y = 0,0

        ## 找到车的位置
        for i in range(8):
            for j in range(8):
                if board[i][j] == 'R':
                    x = i
                    y = j
                    break
    
        ## 左
        for i in range(y-1,-1,-1):
            if board[x][i] == 'B':
                break
            if board[x][i] == 'p':
                ans += 1
                break
        # 右
        for i in range(y+1,8):
            if board[x][i] == 'B':
                break
            if board[x][i] == 'p':
                ans += 1
                break

        ## 上
        for i in range(x-1,-1,-1):
            if board[i][y] == 'B':
                break
            if board[i][y] == 'p':
                ans += 1
                break

        ## 下
        for i in range(x+1,8):
            if board[i][y] == 'B':
                break
            if board[i][y] == 'p':
                ans += 1
                break
        return ans

时间复杂度 $O(n^2)$ ，空间复杂度 $O (1)$
在这里插入图片描述

官方解答

使用方向数组，具体参考：官方解答

class Solution:
    def numRookCaptures(self, board: List[List[str]]) -> int:
        cnt, st, ed = 0, 0, 0
        dx, dy = [0, 1, 0, -1], [1, 0, -1, 0]
        for i in range(8):
            for j in range(8):
                if board[i][j] == "R":
                    st, ed = i, j
        for i in range(4):
            step = 0
            while True:
                tx = st + step * dx[i]
                ty = ed + step * dy[i]
                if tx < 0 or tx >= 8 or ty < 0 or ty >= 8 or board[tx][ty] == "B":
                    break
                if board[tx][ty] == "p":
                    cnt += 1
                    break
                step += 1
        return cnt