Dijkstra算法c++
时间: 2023-12-02 16:40:43 浏览: 68
以下是使用C++实现Dijkstra算法的示例代码:
```c++
#include <iostream>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
const int INF = 0x3f3f3f3f; // 定义一个无穷大的常量
struct Edge {
int to, cost;
Edge(int to, int cost) : to(to), cost(cost) {}
};
typedef pair<int, int> P; // 定义一个pair类型,first表示最短距离,second表示顶点编号
vector<Edge> G[1000]; // 图的邻接表表示
int d[1000]; // 存储起点到各个顶点的最短距离
void dijkstra(int s) {
priority_queue<P, vector<P>, greater<P>> que; // 小根堆
memset(d, INF, sizeof(d)); // 初始化距离数组
d[s] = 0; // 起点到自身的距离为0
que.push(make_pair(0, s)); // 将起点加入队列中
while (!que.empty()) {
P p = que.top(); que.pop();
int v = p.second;
if (d[v] < p.first) continue; // 如果当前顶点的最短距离已经被更新,则跳过
for (int i = 0; i < G[v].size(); i++) {
Edge e = G[v][i];
if (d[e.to] > d[v] + e.cost) { // 如果通过当前顶点可以使距离更短
d[e.to] = d[v] + e.cost; // 更新距离
que.push(make_pair(d[e.to], e.to)); // 将更新后的顶点加入队列中
}
}
}
}
int main() {
int n, m, s;
cin >> n >> m >> s; // 输入顶点数、边数和起点编号
for (int i = 0; i < m; i++) {
int u, v, w;
cin >> u >> v >> w; // 输入一条边的起点、终点和权值
G[u].push_back(Edge(v, w)); // 将边加入邻接表中
}
dijkstra(s); // 求解起点到各个顶点的最短距离
for (int i = 0; i < n; i++) {
cout << d[i] << " "; // 输出起点到各个顶点的最短距离
}
return 0;
}
```
阅读全文