聊一聊利用 Dijkstra 求有向图的最短路径
0x00 前言
我们都知道求最短路径有很多方法,比如 Dijkstra 算法、Bellman-Ford 算法、Floyd-Warshall 算法等等,这些算法各有优缺点,其中 Floyd-Warshall 算法时间复杂度较高,但是编码复杂度较小,而 Bellman-Ford 算法适用于处理有负权边的情况。至于本文要讲的 Dijkstra 算法,优点就是时间复杂度较小,但是不能处理有负权边的图。
我们需要根据不同情况选择不同的算法。
0x01 代码分析
#include<stdio.h>
#define MAXVEX 10
int graph[MAXVEX][MAXVEX];
int short_path[MAXVEX];
int prev_v[MAXVEX];
复制代码
先导入标准输入输出库,我们将节点数设置为 10 个,然后设置存储图的 graph、存储源点到其他各点的最短路径长度的 short_path,还有存储最短路径上各节点的前置节点的 prev_v。
1. 建图
void create_graph( void ){
for(int i=0;i<10;i++){
for(int j=0;j<10;j++){
if(i==j){
graph[i][j]=0;
continue;
}
graph[i][j] = 99999;
}
}
printf("请输入有向边的条数:");
int n;
scanf("%d",&n);
for(int k=0;k<n;k++){
int i,j,value;
scanf("%d %d %d",&i,&j,&value);
graph[i][j]=value;
}
}
复制代码
然后就是将图初始化了。这里我们将没个顶点到自己的距离初始化为 0,然后将其他点先全部初始化为 99999,然后在给每条边赋权值。
2. 求最短路径
void Dijkstra(int g[10][10], int v0) {
int final[MAXVEX];
int j = 0;
int k = 0;
for(j = 0; j<MAXVEX; j++) {
final[j] = 0;
short_path[j] = g[v0][j];
prev_v[j] = 0;
}
final[v0] = 1;
short_path[v0] = 0;
for(int i = 1; i < MAXVEX; i++) {
int min = MAXVEX;
k = 0;
for(int j=0; j<MAXVEX; j++) {
if(final[j]==0 && short_path[j]<min) {
min = short_path[j];
k = j;
}
}
final[k] = 1;
for(int j=0; j<MAXVEX; j++) {
if(final[j]==0 && min+g[k][j] < short_path[j]) {
short_path[j] = min+g[k][j];
prev_v[j] = k;
}
}
}
}
复制代码
大餐来了,迪杰斯特拉算法来了!
该函数传入的参数是图和源点。
我们使用一个数组来标记是否源点到该点已经找出了最短路径。
再将 final、short_path、prev_v 初始化,将 v0 的所有边加入 short_path。
因为源点到源点的最短距离已经确定了是 0,所以我们将 final[v0]标记为 1,short_path[v0]标记为 0。
然后接下来的第一个循环表示一共需要找 MAXVEX-1 条最短路径,然后从源点到剩余未被标记的顶点中寻找最短的一条路径。然后将找到的下一个点标记,表示源点到该点的路径已经找到。然后看一看源点到其它各顶点(尚未确定最短路径的顶点)的最短路径中,如果途经顶点 k 的路径长度比原路径更短,就更新。并设置 k 为 j 的前置节点。
3. 输出最短路径
void show_shortest_path( int source ,int end) {
int que[10];
int tot = 1;
printf("source到%d的最短路径: ",end);
que[tot] = end;
tot++;
int i = prev_v[end];
while(i != source) {
que[tot] = i;
tot++;
i = prev_v[i];
}
que[tot] = source;
for(int i=tot;i>=1;i--){
if(i!=1){
printf("%d->",que[i]);
}else{
printf("%d",que[i]);
}
}
printf("\n");
}
复制代码
就是建一个数组当作队列,然后将每次找到的节点放到队列中,然后顺序遍历。
评论