leetcode-542 01矩阵 Python

最新推荐文章于 2021-10-11 16:34:11 发布

Xenonon

最新推荐文章于 2021-10-11 16:34:11 发布

阅读量305

点赞数

文章标签： python leetcode 算法列表 matrix

本文链接：https://blog.csdn.net/Xenonon/article/details/105525885

版权

本文介绍了如何使用Python解决LeetCode中的542题——01矩阵问题。通过深入解析算法思路，展示了如何遍历矩阵并更新其元素，最终得到所需的01转换矩阵。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution:
    def updateMatrix(self, matrix: List[List[int]]) -> List[List[int]]:
        res = [[None for _ in range(len(matrix[0]))] for _ in range(len(matrix))]  # 设定结果集
        q = collections.deque()
        for i in range(len(matrix)):
            for j in range(len(matrix[0])):
                if matrix[i][j] == 0:
                    res[i][j] = 0
                    q.append([i, j])
        while q:
            x, y = q.popleft()
            for x_bias, y_bias in [[0, 1], [0, -1], [1, 0], [-1, 0]]:
                new_x = x + x_bias
                new_y = y + y_bias
                if 0 <= new_x < len(matrix) and 0 <= new_y < len(matrix[0]) and res[new_x][new_y] == None:
                    res[new_x][new_y] = res[x][y] + 1
                    q.append([new_x, new_y])
        return res