LeetCode-Python-1030. 距离顺序排列矩阵单元格

最新推荐文章于 2021-12-24 02:54:08 发布

暴躁老哥在线刷题

最新推荐文章于 2021-12-24 02:54:08 发布

阅读量769

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/qq_32424059/article/details/89430654

Leetcode 同时被 2 个专栏收录

821 篇文章

订阅专栏

Python

664 篇文章

订阅专栏

本文介绍了一种算法，用于将矩阵中的单元格按照与指定点的距离进行排序。通过使用BFS（宽度优先搜索），算法可以确保单元格按曼哈顿距离从小到大的顺序排列。示例展示了不同输入下算法的输出，以及如何处理各种边界情况。

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

给出 R 行 C 列的矩阵，其中的单元格的整数坐标为 (r, c)，满足 0 <= r < R 且 0 <= c < C。

另外，我们在该矩阵中给出了一个坐标为 (r0, c0) 的单元格。

返回矩阵中的所有单元格的坐标，并按到 (r0, c0) 的距离从最小到最大的顺序排，其中，两单元格(r1, c1) 和 (r2, c2) 之间的距离是曼哈顿距离，|r1 - r2| + |c1 - c2|。（你可以按任何满足此条件的顺序返回答案。）

示例 1：

输入：R = 1, C = 2, r0 = 0, c0 = 0
输出：[[0,0],[0,1]]
解释：从 (r0, c0) 到其他单元格的距离为：[0,1]

示例 2：

输入：R = 2, C = 2, r0 = 0, c0 = 1
输出：[[0,1],[0,0],[1,1],[1,0]]
解释：从 (r0, c0) 到其他单元格的距离为：[0,1,1,2]
[[0,1],[1,1],[0,0],[1,0]] 也会被视作正确答案。

示例 3：

输入：R = 2, C = 3, r0 = 1, c0 = 2
输出：[[1,2],[0,2],[1,1],[0,1],[1,0],[0,0]]
解释：从 (r0, c0) 到其他单元格的距离为：[0,1,1,2,2,3]
其他满足题目要求的答案也会被视为正确，例如 [[1,2],[1,1],[0,2],[1,0],[0,1],[0,0]]。

提示：

1 <= R <= 100
1 <= C <= 100
0 <= r0 < R
0 <= c0 < C

思路：

BFS，因为需要按照与点（r0, c0）的距离，一层一层向外扩散。一开始想歪了用了DFS……

class Solution(object):
    def allCellsDistOrder(self, R, C, r0, c0):
        """
        :type R: int
        :type C: int
        :type r0: int
        :type c0: int
        :rtype: List[List[int]]
        """
        dx = [1, -1, 0, 0]
        dy = [0, 0, -1, 1]
        
        res = [[r0, c0]]
        queue = res[:]
        visited = [[0 for i in range(101)] for j in range(101)]
        visited[r0][c0] = 1
        while(queue):
            next_queue = list()
            for node in queue:
                x0, y0 = node[0], node[1]
                
                for k in range(4):
                    x = x0 + dx[k]
                    y = y0 + dy[k]

                    if x < 0 or x >= R or y < 0 or y >= C: #不满足题目条件
                        continue
                    if visited[x][y] == 1: #这个点已经来过了
                        continue
                   
                    res.append([x,y])
                    visited[x][y] = 1
                    next_queue.append([x,y])
                    
            queue = next_queue[:]
 
        return res