Leetcode 317. Shortest Distance from All Buildings （python+cpp）

最新推荐文章于 2022-11-06 10:31:01 发布

努利！奋斗！

最新推荐文章于 2022-11-06 10:31:01 发布

阅读量1.6k

点赞数

本文链接：https://blog.csdn.net/qq_37821701/article/details/108906696

版权

这篇博客主要介绍了LeetCode第317题的解决方案，使用了两种方法：一是暴力求解的BFS算法，从每个空地出发计算到所有建筑物的最短距离之和；二是优化方法，从建筑物出发反向计算，减少搜索空间，最后找到最小总距离。

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

Leetcode 317. Shortest Distance from All Buildings

题目
解法1：BFS
解法2：
二刷C++版本

题目

在这里插入图片描述

解法1：BFS

brutal force的从每一个空地出发，找与building的最短距离之和

class Solution:
    def shortestDistance(self, grid: List[List[int]]) -> int:
        def helper(i,j):
            visited = set()
            buildings = set()
            q = collections.deque()
            q.append((i,j,0))
            visited.add((i,j))
            total_step = 0
            dirs = [[0,1],[0,-1],[-1,0],[1,0]]
            
            while q:
                i,j,step = q.popleft()
                if grid[i][j] == 1 and (i,j) not in buildings:
                    total_step += step
                    buildings.add((i,j))
                if len(buildings) == num_buildings:
                    break
                if grid[i][j]!=1:
                    for d in dirs:
                        x = i+d[0]
                        y = j+d[1]

                        if 0<=x<m and 0<=y<n and (x,y) not in visited and grid[x][y]!=2:
                            q.append((x,y,step+1))
                            visited.add((x,y))
            return total_step if len(buildings)==num_buildings else -1
        
        m,n = len(grid),

最低0.47元/天解锁文章