【LeetCode】无权图的最短路精选7题——单源、多源

目录

无权图的单源最短路问题:

[1. 迷宫中离入口最近的出口(中等)](#1. 迷宫中离入口最近的出口(中等))

[2. 最小基因变化(中等)](#2. 最小基因变化(中等))

[3. 单词接龙(困难)](#3. 单词接龙(困难))

[4. 为高尔夫比赛砍树(困难)](#4. 为高尔夫比赛砍树(困难))

无权图的多源最短路问题:

[1. 01矩阵(中等)](#1. 01矩阵(中等))

[2. 地图中的最高点(中等)](#2. 地图中的最高点(中等))

[3. 地图分析(中等)](#3. 地图分析(中等))


无权图的单源最短路问题:

使用单源BFS可以求解无权图的单源最短路问题,这是由BFS总是按照距离由近到远来遍历图中每个顶点的性质决定的。

1. 迷宫中离入口最近的出口(中等)

cpp 复制代码
class Solution {
public:
    int nearestExit(vector<vector<char>>& maze, vector<int>& entrance) {
        int m = maze.size();
        int n = maze[0].size();
        vector<vector<bool>> visited(m, vector<bool>(n));
        int ans = 0;

        // 坐标上下左右的偏移量
        int dr[4] = { -1,1,0,0 };
        int dc[4] = { 0,0,-1,1 };

        queue<pair<int, int>> q;
        // 起始坐标入队并标记
        q.push({entrance[0], entrance[1]});
        visited[entrance[0]][entrance[1]] = true;
        while (!q.empty())
        {
            // 要向外拓展一层,步数++
            ans++;
            int count = q.size(); // 本层坐标点的个数
            for (int i = 0; i < count; i++)
            {
                // 队头出队
                auto [row, col] = q.front();
                q.pop();
                // 判断刚才出队的坐标上下左右是否满足条件,再判断是否到达出口,到达则直接返回,没到达则入队并标记
                for (int j = 0; j < 4; j++)
                {
                    int r = row + dr[j];
                    int c = col + dc[j];
                    if (r >= 0 && r < m && c >= 0 && c < n && maze[r][c] == '.' && !visited[r][c])
                    {
                        if (r == 0 || r == m - 1 || c == 0 || c == n - 1) // 到达出口
                            return ans;
                        q.push({r, c});
                        visited[r][c] = true;
                    }
                }
            }
        }
        return -1;
    }
};

2. 最小基因变化(中等)

将距离为1(变化了一个字符)的字符串连接起来构成一个无权图,求start到end的最短路。

cpp 复制代码
class Solution {
public:
    int minMutation(string startGene, string endGene, vector<string>& bank) {
        unordered_set<string> hash(bank.begin(), bank.end()); // 哈希表记录基因库中的字符串
        unordered_set<string> visited; // 标记字符串是否被访问过
        string change = "ACGT";

        if (startGene == endGene)
            return 0;
        if (!hash.count(endGene))
            return -1;

        int ans = 0;
        queue<string> q;
        // 起始字符串入队并标记
        q.push(startGene);
        visited.insert(startGene);
        while (!q.empty())
        {
            // 要向外拓展一层,步数++
            ans++;
            int count = q.size(); // 本层字符串的个数
            for (int i = 0; i < count; i++)
            {
                // 队头出队
                string cur = q.front();
                q.pop();
                // 判断刚才出队的字符串只改变一个字符后是否满足条件,再判断是否到达end,到达则直接返回,没到达则入队并标记
                for (int i = 0; i < 8; i++)
                {
                    string tmp = cur;
                    for (int j = 0; j < 4; j++)
                    {
                        tmp[i] = change[j];
                        if (hash.count(tmp) && !visited.count(tmp))
                        {
                            if (tmp == endGene) // 到达end
                                return ans;
                            q.push(tmp);
                            visited.insert(tmp);
                        }
                    }
                }
            }
        }
        return -1;
    }
};

3. 单词接龙(困难)

和上一题"最小基因变化"类似,区别是上一题求步数,本题求最短路的顶点数。

cpp 复制代码
class Solution {
public:
    int ladderLength(string beginWord, string endWord, vector<string>& wordList) {
        unordered_set<string> hash(wordList.begin(), wordList.end()); // 哈希表记录字典中的单词
        unordered_set<string> visited; // 标记单词是否被访问过

        if (!hash.count(endWord))
            return 0;

        int ans = 1;
        queue<string> q;
        // 起始单词入队并标记
        q.push(beginWord);
        visited.insert(beginWord);
        while (!q.empty())
        {
            // 要向外拓展一层,顶点数++
            ans++;
            int count = q.size(); // 本层单词的个数
            for (int i = 0; i < count; i++)
            {
                // 队头出队
                string cur = q.front();
                q.pop();
                // 判断刚才出队的单词只改变一个字符后是否满足条件,再判断是否到达end,到达则直接返回,没到达则入队并标记
                for (int i = 0; i < cur.size(); i++)
                {
                    string tmp = cur;
                    for (char ch = 'a'; ch <= 'z'; ch++)
                    {
                        tmp[i] = ch;
                        if (hash.count(tmp) && !visited.count(tmp))
                        {
                            if (tmp == endWord) // 到达end
                                return ans;
                            q.push(tmp);
                            visited.insert(tmp);
                        }
                    }
                }
            }
        }
        return 0;
    }
};

4. 为高尔夫比赛砍树(困难)

1 -> 2 -> 3 -> 4 -> 5 -> 6 -> 7

给树按从小到大排序,求顺序相邻的树的最短路。

cpp 复制代码
class Solution {
public:
    int cutOffTree(vector<vector<int>>& forest) {
        m = forest.size();
        n = forest[0].size();

        // 给树按从小到大排序,确定砍树的顺序
        vector<pair<int, int>> trees;
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if (forest[i][j] > 1)
                {
                    trees.push_back({i, j});
                }
            }
        }
        sort(trees.begin(), trees.end(), [&](const pair<int, int>& p1, const pair<int, int>& p2)
        {
            return forest[p1.first][p1.second] < forest[p2.first][p2.second];
        });

        // 求顺序相邻的树的最短路
        int beginr = 0;
        int beginc = 0;
        int ans = 0;
        for (auto& [endr, endc] : trees)
        {
            int step = bfs(forest, beginr, beginc, endr, endc);
            if (step == -1)
                return -1;
            ans += step;
            beginr = endr;
            beginc = endc;
        }
        return ans;
    }

private:
    int bfs(vector<vector<int>>& forest, int beginr, int beginc, int endr, int endc)
    {
        if (beginr == endr && beginc == endc)
            return 0;
        
        vector<vector<bool>> visited(m, vector<bool>(n)); // 标记坐标是否被访问过
        int ans = 0;

        queue<pair<int, int>> q;
        // 起始坐标入队并标记
        q.push({beginr, beginc});
        visited[beginr][beginc] = true;
        while (!q.empty())
        {
            // 要向外拓展一层,步数++
            ans++;
            int count = q.size(); // 本层坐标点的个数
            for (int i = 0; i < count; i++)
            {
                // 队头出队
                auto [row, col] = q.front();
                q.pop();
                // 判断刚才出队的坐标上下左右是否满足条件,再判断是否到达end,到达则直接返回,没到达则入队并标记
                for (int j = 0; j < 4; j++)
                {
                    int r = row + dr[j];
                    int c = col + dc[j];
                    if (r >= 0 && r < m && c >= 0 && c < n && forest[r][c] && !visited[r][c])
                    {
                        if (r == endr && c == endc) // 到达end
                            return ans;
                        q.push({r, c});
                        visited[r][c] = true;
                    }
                }
            }
        }
        return -1;
    }

    // 坐标上下左右的偏移量
    int dr[4] = { -1,1,0,0 };
    int dc[4] = { 0,0,-1,1 };
    
    int m;
    int n;
};

无权图的多源最短路问题:

使用多源BFS可以求解无权图的多源最短路问题,将所有的源点当成一个"超级源点",问题就变成了单源最短路问题。

1. 01矩阵(中等)

以所有的0为源点开始BFS。

距离数组dist最开始全部初始化为-1,表示没有被访问过,后续要修改成距离(步数)。所以,首先,我们不用创建visited数组标记坐标有没有被访问过;其次,不用创建step变量记录步数,并且不用计算本层坐标点的个数count,也就是说不用确定坐标点是在第几层。

cpp 复制代码
class Solution {
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)); // 全部初始化为-1

        // 坐标上下左右的偏移量
        int dr[4] = { -1,1,0,0 };
        int dc[4] = { 0,0,-1,1 };
        
        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.empty())
        {
            // 队头出队
            auto [row, col] = q.front();
            q.pop();
            // 判断刚才出队的坐标上下左右是否满足条件,满足条件则入队并修改距离
            for (int j = 0; j < 4; j++)
            {
                int r = row + dr[j];
                int c = col + dc[j];
                if (r >= 0 && r < m && c >= 0 && c < n && dist[r][c] == -1)
                {
                    q.push({r, c});
                    dist[r][c] = dist[row][col] + 1;
                }
            }
        }
        return dist;
    }
};

2. 地图中的最高点(中等)

水域格子的值是确定的,为0,以所有的0为源点开始BFS,和上一题"01矩阵"一模一样。

cpp 复制代码
class Solution {
public:
    vector<vector<int>> highestPeak(vector<vector<int>>& isWater) {
        int m = isWater.size();
        int n = isWater[0].size();
        vector<vector<int>> dist(m, vector<int>(n, -1)); // 全部初始化为-1

        // 坐标上下左右的偏移量
        int dr[4] = { -1,1,0,0 };
        int dc[4] = { 0,0,-1,1 };
        
        queue<pair<int, int>> q;
        // 所有的源点入队并修改距离为0
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if (isWater[i][j] == 1)
                {
                    q.push({i, j});
                    dist[i][j] = 0;
                }
            }
        }
        while (!q.empty())
        {
            // 队头出队
            auto [row, col] = q.front();
            q.pop();
            // 判断刚才出队的坐标上下左右是否满足条件,满足条件则入队并修改距离
            for (int j = 0; j < 4; j++)
            {
                int r = row + dr[j];
                int c = col + dc[j];
                if (r >= 0 && r < m && c >= 0 && c < n && dist[r][c] == -1)
                {
                    q.push({r, c});
                    dist[r][c] = dist[row][col] + 1;
                }
            }
        }
        return dist;
    }
};

3. 地图分析(中等)

以所有的陆地单元格为源点开始BFS,陆地单元格对应在距离数组dist中的值为0,和"01矩阵"、"地图中的最高点"都是一样的题。

cpp 复制代码
class Solution {
public:
    int maxDistance(vector<vector<int>>& grid) {
        int m = grid.size();
        int n = grid[0].size();
        vector<vector<int>> dist(m, vector<int>(n, -1)); // 全部初始化为-1
        int ans = -1;

        // 坐标上下左右的偏移量
        int dr[4] = { -1,1,0,0 };
        int dc[4] = { 0,0,-1,1 };
        
        queue<pair<int, int>> q;
        // 所有的源点入队并修改距离为0
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if (grid[i][j] == 1)
                {
                    q.push({i, j});
                    dist[i][j] = 0;
                }
            }
        }
        while (!q.empty())
        {
            // 队头出队
            auto [row, col] = q.front();
            q.pop();
            // 判断刚才出队的坐标上下左右是否满足条件,满足条件则入队并修改距离
            for (int j = 0; j < 4; j++)
            {
                int r = row + dr[j];
                int c = col + dc[j];
                if (r >= 0 && r < m && c >= 0 && c < n && dist[r][c] == -1)
                {
                    q.push({r, c});
                    dist[r][c] = dist[row][col] + 1;
                    ans = max(ans, dist[r][c]);
                }
            }
        }
        return ans;
    }
};
相关推荐
柠檬少少开发22 分钟前
图像拼接算法及实现(一)
人工智能·算法·计算机视觉
jnrjian32 分钟前
USE_CONCAT in list OR 以及 filter Nest LOOP
数据结构·list
阿华的代码王国39 分钟前
【JavaEE】多线程编程引入——认识Thread类
java·开发语言·数据结构·mysql·java-ee
weixin_486681141 小时前
C++系列-STL容器中统计算法count, count_if
开发语言·c++·算法
一道秘制的小菜1 小时前
C++第七节课 运算符重载
服务器·开发语言·c++·学习·算法
代码小狗Codog2 小时前
C++独立开发开源大数计算库 CBigNum
数据结构·c++
咕噜咕嘟嘟嘟3 小时前
343. 整数拆分
数据结构·算法
WenGyyyL3 小时前
力扣最热一百题——二叉树的直径
java·c++·算法·二叉树·深度优先
sdlkjaljafdg3 小时前
vector<bool>性能测试
开发语言·c++·算法
imaima6663 小时前
数据结构——二叉树的介绍
数据结构