Leetcode -- 2373. 矩阵中的局部最大值

最新推荐文章于 2024-09-02 16:50:54 发布

猫&九

最新推荐文章于 2024-09-02 16:50:54 发布

阅读量76

点赞数

分类专栏： leetcode 刷题文章标签： leetcode 矩阵算法

本文链接：https://blog.csdn.net/qq_47714288/article/details/129289857

版权

leetcode 刷题专栏收录该内容

3 篇文章 0 订阅

订阅专栏

该代码实现了一个名为largestLocal的函数，用于在一个二维网格grid中找到所有3x3子矩阵的最大值，并返回这些最大值构成的矩阵。它首先通过两层循环遍历grid，然后计算每个3x3子矩阵的最大值，并将结果存储在新的答案矩阵ans中。

摘要由CSDN通过智能技术生成

解题

class Solution:
    def largestLocal(self, grid: list[list[int]]) -> list[list[int]]:
        # li = []
        # # 纵向移动
        # for i in range(len(grid)):
        #     tem = []
        #     # 横向移动
        #     for j in range(len(grid)):
        #         # 是否可以构成 3 * 3 的矩阵
        #         if i + 2 < len(grid) and j + 2 < len(grid):
        #             # 3 * 3 矩阵中找出最大值
        #             tem.append(max(grid[i][j], grid[i][j + 1], grid[i][j + 2],
        #                            grid[i + 1][j], grid[i + 1][j + 1], grid[i + 1][j + 2],
        #                            grid[i + 2][j], grid[i + 2][j + 1], grid[i + 2][j + 2]))
        #     if len(tem) != 0:
        #         li.append(tem)
        # return li

        # 第二种
        n = len(grid)
        # 生成结果矩阵，用 0 填充, [0] * (n -2) 一行(n - 2) 个元素，共有 for _ in range(n-2) 行
        ans = [[0] * (n - 2) for _ in range(n - 2)]
        # 纵向移动
        for i in range(n - 2):
            # 横向移动
            for j in range(n - 2):
                # 找出 3 * 3 矩阵中的最大值
                ans[i][j] = max(grid[x][y] for x in range(i, i + 3) for y in range(j, j + 3))
        return ans


s = Solution()
l = [[9, 9, 8, 1], [5, 6, 2, 6], [8, 2, 6, 4], [6, 2, 2, 2]]
print(s.largestLocal(l))