Dijkstra算法代码模板（C++）

Dijkstra算法是一种用于寻找带权图中从一个顶点到其它顶点的最短路径的算法。C++类模板可以实现该算法的优化，在模板中可以使用泛型编程的思想进行算法的扩展，使得代码更加灵活和可重用。以下是Dijkstra算法的类模板实现示例： ```c++ #include <iostream> #include <vector> #include <queue> #include <functional> using namespace std; template<typename T> class Graph { private: int V; // 顶点个数 vector<vector<pair<int, T>>> adj; // 邻接表 public: Graph(int V) { this->V = V; adj.resize(V); } void addEdge(int u, int v, T weight) { adj[u].push_back({v, weight}); adj[v].push_back({u, weight}); } // Dijkstra算法 vector<T> dijkstra(int start) { priority_queue<pair<T, int>, vector<pair<T, int>>, greater<pair<T, int>>> pq; vector<T> dist(V, numeric_limits<T>::max()); vector<bool> visited(V, false); dist[start] = 0; pq.push({0, start}); while (!pq.empty()) { int u = pq.top().second; pq.pop(); if (visited[u]) continue; visited[u] = true; for (auto v : adj[u]) { int vertex = v.first; T weight = v.second; if (!visited[vertex] && dist[u] + weight < dist[vertex]) { dist[vertex] = dist[u] + weight; pq.push({dist[vertex], vertex}); } } } return dist; } }; int main() { Graph<int> g(4); g.addEdge(0, 1, 5); g.addEdge(0, 2, 1); g.addEdge(1, 2, 2); g.addEdge(1, 3, 6); g.addEdge(2, 3, 4); vector<int> dist = g.dijkstra(0); for (int i = 0; i < dist.size(); i++) cout << "Vertex " << i << " has shortest distance " << dist[i] << "\n"; return 0; } ``` 该示例中用一个邻接表来存储图中的边，每个邻接表项表示一个由顶点索引和权重组成的有向边。dijkstra()函数实现了具有优先级队列的Dijkstra算法，其中使用visited向量表示已访问过的顶点，dist向量表示从起始点到每个顶点的最短路径长度。最后，将dist向量返回以供其他代码使用。