如何使用C++中的最短路径算法

如何使用C++中的最短路径算法

最短路径算法是图论中的关键算法之一，它用来确定两个顶点之间的最短路径。在C++语言中，提供了许多实现最短路径算法的库，例如Dijkstra算法和Floyd-Warshall算法。本文将为您详细介绍如何使用这两种算法，并提供相应的代码示例。

1. Dijkstra算法

Dijkstra算法是一种贪心算法，用于解决带权有向图中单源最短路径问题。下面是使用C++语言实现Dijkstra算法的代码示例：

#include <iostream>
#include <vector>
#include <queue>

const int INF = 1e9;

void dijkstraAlgorithm(int start, const std::vector<std::vector<std::pair<int, int>>>& graph, std::vector<int>& distance) {
    int n = graph.size();
    distance.resize(n, INF);
    distance[start] = 0;

    std::priority_queue<std::pair<int, int>, std::vector<std::pair<int, int>>, std::greater<std::pair<int, int>>> pq;
    pq.push({0, start});

    while (!pq.empty()) {
        int u = pq.top().second;
        int dist = pq.top().first;
        pq.pop();

        if (dist > distance[u]) {
            continue;
        }

        for (const auto& neighbor : graph[u]) {
            int v = neighbor.first;
            int weight = neighbor.second;

            if (distance[u] + weight < distance[v]) {
                distance[v] = distance[u] + weight;
                pq.push({distance[v], v});
            }
        }
    }
}

int main() {
    int n, m;
    std::cin >> n >> m;

    std::vector<std::vector<std::pair<int, int>>> graph(n);

    for (int i = 0; i < m; ++i) {
        int u, v, w;
        std::cin >> u >> v >> w;
        graph[u].push_back({v, w});
    }

    int start;
    std::cin >> start;

    std::vector<int> distance;
    dijkstraAlgorithm(start, graph, distance);

    for (int i = 0; i < n; ++i) {
        std::cout << "Distance from " << start << " to " << i << " is " << distance[i] << std::endl;
    }

    return 0;
}

以上代码实现了Dijkstra算法。首先，从输入中读取图的结点数n和边数m。然后，创建一个邻接表来表示图的结构，并将边的信息存储在邻接表中。接下来，读取起始结点start。最后，调用dijkstraAlgorithm函数计算从起始结点到其他结点的最短路径，并输出结果。

Unscreen

AI智能视频背景移除工具

下载

立即学习“C++免费学习笔记（深入）”；

2. Floyd-Warshall算法

Floyd-Warshall算法用于解决带权有向图中所有顶点之间的最短路径问题。下面是使用C++语言实现Floyd-Warshall算法的代码示例：

#include <iostream>
#include <vector>

const int INF = 1e9;

void floydWarshallAlgorithm(const std::vector<std::vector<int>>& graph, std::vector<std::vector<int>>& distance) {
    int n = graph.size();

    distance = graph;

    for (int k = 0; k < n; ++k) {
        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < n; ++j) {
                if (distance[i][k] != INF && distance[k][j] != INF && distance[i][k] + distance[k][j] < distance[i][j]) {
                    distance[i][j] = distance[i][k] + distance[k][j];
                }
            }
        }
    }
}

int main() {
    int n, m;
    std::cin >> n >> m;

    std::vector<std::vector<int>> graph(n, std::vector<int>(n, INF));

    for (int i = 0; i < m; ++i) {
        int u, v, w;
        std::cin >> u >> v >> w;
        graph[u][v] = w;
    }

    std::vector<std::vector<int>> distance;
    floydWarshallAlgorithm(graph, distance);

    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            if (distance[i][j] == INF) {
                std::cout << "No path from " << i << " to " << j << std::endl;
            } else {
                std::cout << "Distance from " << i << " to " << j << " is " << distance[i][j] << std::endl;
            }
        }
    }

    return 0;
}

以上代码实现了Floyd-Warshall算法。首先，从输入中读取图的结点数n和边数m。然后，创建一个邻接矩阵来表示图的结构，并将边的信息存储在邻接矩阵中。最后，调用floydWarshallAlgorithm函数计算所有顶点之间的最短路径，并输出结果。

通过以上代码示例，您可以学习如何在C++中使用Dijkstra算法和Floyd-Warshall算法来解决最短路径问题。希望本文能对您有所帮助并增加您对最短路径算法的理解。