leetcode：317. 离建筑物最近的距离

题目描述

• leetcode：317-vip. 离建筑物最近的距离 shortest-distance-from-all-buildings
• lintcode：803 · 建筑物之间的最短距离

题目解析

class Solution{
public:
    int shortestDistance(vector<vector<int>>& grid){
        
    }
};

题目解析

• 因为要求最短距离，所以应该用BFS而不是DFS，这样更加简单些
• BFS需要借助queue，我们对于每一个建筑的位置都进行一次全图的BFS遍历，每次都建立一个dist的距离场

class Solution{
    std::vector<std::vector<int>> directions {{0, -1}, {0, 1}, {-1, 0}, {1, 0}};


    int bfs(vector<vector<int>>& grid, int m, int n, int x, int y, int mark, std::vector<std::vector<int>>& totalDist ){
        int ans = INT_MAX;
        std::queue<std::vector<int>> queue;
        // 队列中每个数组有 3 个元素，分别表示：
        // 第一个元素和第二个元素表示坐标值
        // 第三个元素表示当前坐标到建筑物的距离
        // 第三个元素的初始值为 0 的原因是：一开始的时候从当前建筑物到当前建筑物的距离是 0
        queue.push({x, y, 0});
        while (!queue.empty()){
            auto curPoint = queue.front(); queue.pop();
            int currDist = curPoint[2];
            for(auto direction : directions){
                int newX = curPoint[0] + direction[0];
                int newY = curPoint[1] + direction[1];
                if (newX >= 0 && newX < m && newY >= 0 && newY < n && // 不超出边界
                    grid[newX][newY] == mark) { // 当前点是空地
                    // (2) 在搜索的同时计算每一个空格到这个建筑物的距离
                    int dist = currDist + 1;
                    // (3) 在搜索的同时将每一个空格到每一个建筑物的距离进行累加
                    totalDist[newX][newY] += dist;
                    // (4) 取空格到所有建筑物的最小距离
                    ans = std::min(ans, totalDist[newX][newY]);
                    queue.push({newX, newY, dist});
                    // 当前点处理过了
                    grid[newX][newY]--;
                }
            }
        }
        
        return ans == INT_MAX ? -1 : ans;
    }

public:
    // 思路：
    //  (1) 从每一个建筑物开始进行广度优先搜索
    //  (2) 在搜索的同时计算每一个空格到这个建筑物的距离
    //  (3) 在搜索的同时将每一个空格到每一个建筑物的距离进行累加，得到每个空格到所有建筑物的距离
    //  (4) 取空格到所有建筑物的最小距离
    int shortestDistance(vector<vector<int>>& grid){
        if(grid.empty()|| grid[0].empty()){
            return 0;
        }

        int m = grid.size(), n = grid[0].size();
        std::vector<std::vector<int>> totalDist(m, std::vector<int>(n));
        int ans = INT_MAX;
       //  这里还需要注意一点的是：建筑物个数为 n，因此每一个空格都会被遍历 n，次，因为要计算空格到达每一个建筑物的距离嘛。
        // 用于标记空地
       //  初始时，空格是用 mark 表示，当我计算完一个建筑物时，mark–，所有的空格变成用 -1 表示了。
       //  这样做就是为了解决“BFS时空格不能重复遍历，但是遍历建筑物时，空格又要重复使用”的矛盾。
       int mark = 0;
        for (int i = 0; i < m; ++i) {
            for (int j = 0; j < n; ++j) {
                // (1) 从每一个建筑物开始进行广度优先搜索
                if(grid[i][j] == 1){
                    ans = bfs(grid, m, n, i, j, mark, totalDist);
                    --mark;
                }
            }
        }
        return ans;
    }
};

测试：

int main() {
    vector<vector<int>> rooms
            {{1, 0, 2, 0, 1},
             {0, 0, 0, 0, 0},
             {0, 0, 1, 0, 0},
            };

    Solution a;
    cout << a.shortestDistance(rooms);


    return 0;
}

类似题目

题目	思想
leetcode：286：墙和门 Walls and Gates
leetcode：296.最佳的碰头地点 Best Meeting Point
leetcode：317.离建筑物最近的距离 Shortest Distance from All Buildings