温馨提示×

C++ Dijkstra算法怎样实现图的输入

c++
小樊
82
2024-07-25 17:32:13
栏目: 编程语言

在C++中,可以使用邻接矩阵或邻接表来表示图,并实现Dijkstra算法。以下是使用邻接表表示图的示例代码:

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

using namespace std;

struct Node {
    int dest;
    int weight;
};

class Graph {
private:
    int V;
    vector<vector<Node>> adjList;

public:
    Graph(int vertices) {
        V = vertices;
        adjList.resize(V);
    }

    void addEdge(int src, int dest, int weight) {
        Node node = { dest, weight };
        adjList[src].push_back(node);
    }

    void dijkstra(int src) {
        vector<int> dist(V, INT_MAX);
        dist[src] = 0;

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

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

            for (Node neighbor : adjList[u]) {
                int v = neighbor.dest;
                int weight = neighbor.weight;

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

        cout << "Shortest distances from source " << src << " to other vertices:" << endl;
        for (int i = 0; i < V; i++) {
            cout << "Vertex " << i << ": " << dist[i] << endl;
        }
    }
};

int main() {
    Graph g(6);

    g.addEdge(0, 1, 5);
    g.addEdge(0, 2, 3);
    g.addEdge(1, 3, 6);
    g.addEdge(1, 4, 2);
    g.addEdge(2, 1, 1);
    g.addEdge(2, 4, 4);
    g.addEdge(3, 5, 8);
    g.addEdge(4, 5, 7);

    g.dijkstra(0);

    return 0;
}

在这个示例中,首先创建一个Graph类来表示图,其中包括顶点数量V和邻接表adjList。addEdge函数用于添加边,dijkstra函数用于计算从源顶点到其他顶点的最短路径。

在main函数中,创建一个Graph对象并添加边,然后调用dijkstra函数计算最短路径。输出结果为从源顶点0到其他顶点的最短距离。您可以根据自己的需求修改顶点数量和边的权重来自定义图。

0