【C++算法】BFS解决多源最短路问题相关经典算法题

1.01矩阵

既然本章是BFS解决多源最短路问题，也就是说有若干个起点，那我们就可以暴力一点，直接把多源最短路径问题转化成若干个单源最短路径问题，然后将每次的步数比较一下，取到最短的就是最短路径的结果，这个想法好写，但是在本章都会超时，所以我们换一个思路：把所有的源点当成一个超级源点(包含所有的源点)，那么问题就变成了单一的单源最短路径问题，此时我们使用一次bfs就可以解决问题。但是如何将这个源点当作一个超级源点呢？在解决单源最短路问题，我们的做法是把起点加入到队列，然后一层一层的往外扩展，所以我们只需要一开始将所有的起点都加入到队列，然后一层一层往外扩展即可，我们直接看第一个思路：

代码我已经替大家试过了，会超时，所以我们来看第二种思路：

直接上代码：

class Solution {
    int dx[4] = {0, 0, 1, -1};
    int dy[4] = {1, -1, 0, 0};
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& mat) {
        int m = mat.size();
        int n = mat[0].size();
        vector<vector<int>> dist(m, vector<int>(n, -1));

        // dist[i][j] == -1,表示:没有被搜索过
        // dist[i][j] != -1,表示:最短距离
        // 利用dist来直接标记该位置是否已经被使用
        queue<pair<int,int>> q;
        // 把所有为0的源点加入队列中
        for(int i = 0; i < m; i++)  
            for(int j = 0; j < n; j++)
                if(mat[i][j] == 0)
                {
                    q.push({i, j});
                    dist[i][j] = 0;
                }

        while(q.size())
        {
            // 此时我们不需要同时向外扩展
            // 因此不需要使用sz
            auto [a, b] = q.front();
            q.pop();
            for(int i = 0; i < 4; i++)
            {
                int x = a + dx[i], y = b + dy[i];
                if(x >= 0 && x < m && y >= 0 && y < n && dist[x][y] == -1)
                {
                    // 此时说明已经找到，直接修改dist的值即可
                    dist[x][y] = dist[a][b] + 1;
                    q.push({x, y});
                }
            }
        }
    return dist;
    }
};

2.飞地的数量

这道题目如果我们按照每次遍历数组每次遇到1就去bfs，肯定是会超时的，但是根据这个写法也有不超时的做法，我们遍历的时候，遇到1就去bfs，并将途中遇到1的位置标记一下，下一轮遇到1的时候，先dfs一下，如果这次dfs宽搜的时候遇到1，发现这个位置已经标记过，此时就不要bfs了，直接就是上次的结果，但是这样写需要两次dfs，并且dsf的代码还不一样，也比较麻烦，那我们来一个新思路：正难则反，我们可以从边上的 1 开始搜索，把与边上 1 相连的联通区域全部标记⼀下； 然后再遍历⼀遍矩阵，看看哪些位置的 1 没有被标记即可标记的时候，可以用「多源 bfs 」解决，bfs的逻辑和上一个题目基本差不多，直接上代码。

class Solution {
    int dx[4] = {0, 0, 1, -1};
    int dy[4] = {1, -1, 0, 0};
    bool vis[501][501] = {false};
public:
    int numEnclaves(vector<vector<int>>& grid) {
        int m = grid.size();
        int n = grid[0].size();
         // 1. 把边上的 1 加⼊到队列中
        queue<pair<int,int>> q;
        // 遍历行
        for(int i = 0; i < n; i++)
        {
            if(grid[0][i] == 1)
                q.push({0, i});
            if(grid[m - 1][i] == 1)
                q.push({m - 1, i});
        }
        // 遍历列
        for(int j = 0; j < m; j++)
        {
            if(grid[j][0] == 1)
                q.push({j, 0});
            if(grid[j][n - 1] == 1)
                q.push({j, n - 1});
        }
        // 2. 多源bfs
        while(q.size())
        {
            auto [a, b] = q.front();
            q.pop();
            vis[a][b] = true;
            for(int i = 0; i < 4; i++)
            {
                int x = a + dx[i];
                int y = b + dy[i];
                if(x >= 0 && x < m && y >= 0 && y < n && grid[x][y] == 1 && !vis[x][y])
                {
                    q.push({x, y});
                    vis[x][y] = true;
                }
            }
        }
        // 统计结果
        int ret = 0;
        for(int i = 0; i < m; i++)
            for(int j = 0; j < n; j++)
                if(grid[i][j] == 1 && !vis[i][j])
                    ret++;
        return ret;       
    }
};

3.地图中的最高点

注意本题给的是结果矩阵，我们可以先来看看原始矩阵的样子

此时就一目了然了，我们直接创建一个height数组，将isWater中的水域依次向外扩展，直接转化成多源bfs的问题，并且我们发现它就是01矩阵的变型题，直接上代码了：

class Solution {
    int dx[4] = { 0, 0, 1, -1 };
    int dy[4] = { 1, -1, 0, 0 };
public:
    vector<vector<int>> highestPeak(vector<vector<int>>& isWater) {
        int m = isWater.size();
        int n = isWater[0].size();
        // 数组不能初始化为0,因为数组中存在0 - 代表水域
        vector<vector<int>> height(m, vector<int>(n, -1));

        queue<pair<int, int>> q;
        // 把所有的源点加入到队列里面
        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
                if (isWater[i][j] == 1)
                {
                    height[i][j] = 0;
                    q.push({ i,j });
                }
        // 多源bfs
        while (q.size())
        {
            auto [a, b] = q.front();
            q.pop();
            for (int i = 0; i < 4; i++)
            {
                int x = a + dx[i];
                int y = b + dy[i];
                // 该值没有被搜索过
                if (x >= 0 && x < m && y >= 0 && y < n && height[x][y] == -1)
                {
                    height[x][y] = height[a][b] + 1;
                    q.push({ x, y });
                }
            }
        }

        return height;
    }
};

4.地图分析

我们这个题目要求解海洋到陆地的最大距离，也可以求，但是海洋数量多蛮烦，正难则反，我们可以求陆地到海洋，将陆地作为超级源点依次向外扩展，随后创建一个dist数组，记录每次bfs的值保存在ist数组中即可解决，代码和上面的题目都相似，直接上代码：

class Solution
{
    int dx[4] = { 0, 0, 1, -1 };
    int dy[4] = { 1, -1, 0, 0 };
public:
    int maxDistance(vector<vector<int>>& grid)
    {
        int m = grid.size(), n = grid[0].size();
        vector<vector<int>> dist(m, vector<int>(n, -1));
        queue<pair<int, int>> q;
        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
                if (grid[i][j] == 1)
                {
                    dist[i][j] = 0;
                    q.push({ i, j });
                }
        int ret = -1;
        while (q.size())
        {
            auto [a, b] = q.front(); q.pop();
            for (int i = 0; i < 4; i++)
            {
                int x = a + dx[i], y = b + dy[i];
                if (x >= 0 && x < m && y >= 0 && y < n && dist[x][y] == -1)
                {
                    dist[x][y] = dist[a][b] + 1;
                    q.push({ x, y });
                    ret = max(ret, dist[x][y]);
                }
            }
        }
        return ret;
    }
};