前言
单源点最短路径问题是图论一种经典的问题,一般用Kijkstra算法解决比较多。因为网上大多都是C++写算法并且没有完整的源码,这使得新手需要花更多的时间去理解,所以我下面写一些自己的见解,并给出测试数据。
一、头文件、宏定义、结构体
这里并不需要像堆栈啊队列啊链表啊搞很多,一个一维数组存储顶点,一个二维数组存储邻接矩阵就行。但是先拉出来,方便大家去参考下面的代码。其中INF代表无穷大,因为设置太大main会出现栈溢出,所以我定义100就可以了,其实权重也没有太大的(具体见题目)。
#include<stdio.h>
#include<stdlib.h>
#define MAX_SIZE 100
#define INF 100
struct Matrix
{
int Vertex[MAX_SIZE];//顶点
int Arc[MAX_SIZE][MAX_SIZE];//边
int VertexNum, ArcNum;
};
二、初始化和输入邻接矩阵
初始化非常简单,只需让每个顶点的信息都赋值成INF即可;我习惯做啥都先打印一下,这样是为了方便我检查BUG。至于我的循环总是从1开始,也是个人的习惯,因为很多时候,编号和下标会错位,这样是为了避免出现这种情况。
void GraphInit(struct Matrix* graph)
{
for (int i = 1; i <= graph->VertexNum; i++)
{
for (int j = 1; j <= graph->VertexNum; j++)
{
graph->Arc[i][j] = INF;
}
}
}
void GraphCreate(struct Matrix* graph)
{
int a, b, w;
for (int i = 1; i <= graph->ArcNum; i++)
{
printf("请输入边所依附的两个顶点以及权重的信息:\n");
scanf("%d%d%d", &a, &b, &w);
graph->Arc[a][b] = w;
}
printf("打印邻接矩阵:\n");
for (int i = 1; i <= graph->VertexNum; i++)
{
for (int j = 1; j <= graph->VertexNum; j++)
{
printf("%5d", graph->Arc[i][j]);
}
printf("\n");
}
}
三、Dijsktra算法函数
这里先说一下各个数组的意义:
from是起点,也就是源点,to是终点。
visited数组记录是否访问过,已经访问的赋值为1,初始化时全部赋值为0;
dist数组表示源点到各个顶点的距离,并且我们从这里去找最小距离以更新,初始化赋值源点到各顶点的距离;
path数组表示经过的路径,如果没有经过,那么存储的元素是-1,如果有经过,那么存储这个经过的顶点,举个例子:1通过3到达5,那么5(下标从1开始)这个位置存储的是3。初始化时跟其他不一样哈,因为一开始我们只拿到了源点的信息,那么所有顶点只能由源点from去到达,当然等于INF即没有直接连接的顶点肯定只能赋个-1啦。
下面就是重头戏!!!因为我们不断地在dist数组里面找最小距离,因此dist数组也是循环一次更新一次的。但是必须要注意哈,更新次数并不是顶点个数,因为源点那一次已经初始化过了,所以要少一次捏。同样要更新的是临时变量min,别写在内层循环里面了哈!找的话就没有什么可以说的了。但是更新dist数组那个循环需要好好理解一下:
因为我们找到了新的顶点,那么我们是不是要考虑一下,能否通过这个顶点去更新最短路径,也就是更新dist数组。所以我们得看看新的顶点所对应的边,即查找邻接矩阵Arc[pos][i],而dist[pos]就是上一个顶点到该顶点的距离,二者相加跟原来的dist数组的这个位置比较,如果小于原来的距离,那么更新。
int Dijsktra(int from,int to,int path[],int dist[],struct Matrix*graph)
{
int visited[MAX_SIZE];//访问数组
for (int i = 1; i <= graph->VertexNum; i++)//初始化访问数组
{
visited[i] = 0;//0代表没有访问过
}
for (int i = 1; i <= graph->VertexNum; i++)//初始化距离数组
{
dist[i] = graph->Arc[from][i];//从源点到其他点的距离
}
for (int i = 1; i <= graph->VertexNum; i++)//初始化路径数组
{
if (dist[i] != INF) path[i] = from;
else path[i] = -1;
}
for (int i = 1; i < graph->VertexNum; i++)//对除源点外的每一个点进行最短距离计算
{
int min = INF;//更新最小距离
int pos;//距离最小距离的下标
for (int j = 1; j <= graph->VertexNum; j++)//找到dist数组里面距离最小的
{
if (!visited[j] && dist[j] < min)
{
pos = j;
min = dist[j];
}
}
visited[pos] = 1;//标记为已访问
for (int j = 1; j <= graph->VertexNum; j++)
{
if (!visited[j] && (dist[pos] + graph->Arc[pos][j]) < dist[j])//更新dist数组
{
dist[j] = dist[pos] + graph->Arc[pos][j];
path[j] = pos;
}
}
}
printf("最短路径长度为:\n");
printf("%d\n", dist[to]);
}
四、打印路径
我看了其他博客,很多都是用堆栈写的,因为我看不懂,我自己用递归更加简洁哈。思路大抵是这样的:(来自b站 蓝猫老师 的视频截图)下面上代码,很好理解。
path[path[path[终点]]]...就是这样的递归哈!
但是这样是不能打印终点的,因为path数组里面存的是路径,没有终点。
void PrintPath(int path[], int from,int to)
{
if (path[to] != from)
{
PrintPath(path, from, path[to]);
printf("%d ", path[to]);
printf("->");
}
else
{
printf("%d ", path[to]);
printf("->");
}
}
总结
上完整源码和测试用例!!!
#include<stdio.h>
#include<stdlib.h>
#define MAX_SIZE 100
#define INF 100
struct Matrix
{
int Vertex[MAX_SIZE];//顶点
int Arc[MAX_SIZE][MAX_SIZE];//边
int VertexNum, ArcNum;
};
void GraphInit(struct Matrix* graph)
{
for (int i = 1; i <= graph->VertexNum; i++)
{
for (int j = 1; j <= graph->VertexNum; j++)
{
graph->Arc[i][j] = INF;
}
}
}
void GraphCreate(struct Matrix* graph)
{
int a, b, w;
for (int i = 1; i <= graph->ArcNum; i++)
{
printf("请输入边所依附的两个顶点以及权重的信息:\n");
scanf("%d%d%d", &a, &b, &w);
graph->Arc[a][b] = w;
}
printf("打印邻接矩阵:\n");
for (int i = 1; i <= graph->VertexNum; i++)
{
for (int j = 1; j <= graph->VertexNum; j++)
{
printf("%5d", graph->Arc[i][j]);
}
printf("\n");
}
}
int Dijkstra(int from,int to,int path[],int dist[],struct Matrix*graph)
{
int visited[MAX_SIZE];//访问数组
for (int i = 1; i <= graph->VertexNum; i++)//初始化访问数组
{
visited[i] = 0;//0代表没有访问过
}
for (int i = 1; i <= graph->VertexNum; i++)//初始化距离数组
{
dist[i] = graph->Arc[from][i];//从源点到其他点的距离
}
for (int i = 1; i <= graph->VertexNum; i++)//初始化路径数组
{
if (dist[i] != INF) path[i] = from;
else path[i] = -1;
}
for (int i = 1; i < graph->VertexNum; i++)//对除源点外的每一个点进行最短距离计算
{
int min = INF;//更新最小距离
int pos;//距离最小距离的下标
for (int j = 1; j <= graph->VertexNum; j++)//找到dist数组里面距离最小的
{
if (!visited[j] && dist[j] < min)
{
pos = j;
min = dist[j];
}
}
visited[pos] = 1;//标记为已访问
for (int j = 1; j <= graph->VertexNum; j++)
{
if (!visited[j] && (dist[pos] + graph->Arc[pos][j]) < dist[j])//更新dist数组
{
dist[j] = dist[pos] + graph->Arc[pos][j];
path[j] = pos;
}
}
}
printf("最短路径长度为:\n");
printf("%d\n", dist[to]);
}
void PrintPath(int path[], int from,int to)
{
if (path[to] != from)
{
PrintPath(path, from, path[to]);
printf("%d ", path[to]);
printf("->");
}
else
{
printf("%d ", path[to]);
printf("->");
}
}
int main()
{
struct Matrix graph;
printf("请输入顶点个数和边的个数:\n");
int vertexnum, arcnum;
scanf("%d%d", &graph.VertexNum, &graph.ArcNum);
GraphInit(&graph);
GraphCreate(&graph);
printf("请输入起点和终点:\n");
int from, to;
scanf("%d%d", &from, &to);
int dist[MAX_SIZE];//距离数组
int path[MAX_SIZE];//路径数组
Dijkstra(from, to,path,dist,&graph);
printf("最短路径为:\n");
PrintPath(path,from,to);
printf("%d", to);
return 0;
}