数据结构中迪杰斯特拉算法代码实现
时间: 2023-10-29 15:06:29 浏览: 67
迪杰斯特拉算法代码实现
5星 · 资源好评率100%
迪杰斯特拉算法(Dijkstra's Algorithm)是一种用于解决带权图中最短路径问题的算法。它是一种贪心算法,每次找到距离起点最近的顶点,并更新其它顶点的距离。
下面是一个基于邻接矩阵的迪杰斯特拉算法的代码实现:
```c++
#include <iostream>
#include <limits.h>
using namespace std;
// 定义邻接矩阵大小
#define V 6
// 找到未处理顶点中距离起点最近的顶点
int minDistance(int dist[], bool sptSet[])
{
int min = INT_MAX, min_index;
for (int v = 0; v < V; v++)
if (!sptSet[v] && dist[v] <= min)
min = dist[v], min_index = v;
return min_index;
}
// 打印最短路径
void printSolution(int dist[])
{
printf("Vertex \t Distance from Source\n");
for (int i = 0; i < V; i++)
printf("%d \t\t %d\n", i, dist[i]);
}
void dijkstra(int graph[V][V], int src)
{
int dist[V]; // 存储最短距离
bool sptSet[V]; // 标记是否处理过
for (int i = 0; i < V; i++)
dist[i] = INT_MAX, sptSet[i] = false;
dist[src] = 0;
for (int count = 0; count < V - 1; count++) {
int u = minDistance(dist, sptSet);
sptSet[u] = true;
for (int v = 0; v < V; v++)
if (!sptSet[v] && graph[u][v] && dist[u] != INT_MAX && dist[u] + graph[u][v] < dist[v])
dist[v] = dist[u] + graph[u][v];
}
printSolution(dist);
}
int main()
{
int graph[V][V] = {{0, 2, 0, 6, 0, 0},
{2, 0, 3, 8, 5, 0},
{0, 3, 0, 0, 7, 0},
{6, 8, 0, 0, 9, 10},
{0, 5, 7, 9, 0, 1},
{0, 0, 0, 10, 1, 0}};
dijkstra(graph, 0);
return 0;
}
```
该算法的时间复杂度为 O(V^2),其中 V 为顶点数量。
阅读全文