概述
Dijkstra算法主要用于解决单源最短路径问题。在题目中,一般除了求最短路径还会附加其他条件加以筛选,求得唯一路径。
标准套路
st=起点,en=终点,D[ ]=到各点的最短路径,dis[ ][ ]=各点之间的路径长度(注意,a到b和b到a都要存储),visit[ ]=访问控制,pre[ ]=各点的前驱结点
首先要将不能用0初始化的各对象用fill函数初始化
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int inf=99999;
fill(D,D+505,inf);
fill(dis[0],dis[0]+505*505,inf);
接下来就是dij的主要算法
void dij(){
D[st]=0;
for(int i=0;i<N;i++){
int u=-1,dmin=inf;
for(int j=0;j<N;j++){
if(visit[j]==0&&D[j]<dmin){
dmin=D[j];
u=j;
}
}
if(u==-1) break;
visit[u]=1;
for(int v=0;v<N;v++){
if(visit[v]==0&&dis[u][v]!=inf){
if(D[u]+dis[u][v]<D[v]){
D[v]=D[u]+dis[u][v];
pre[v]=u;
}
}
}
}
}
最后得到的是所有结点的前驱数组pre[]
和最短路径长度数组D[]
,如果需要输出路径,则可以利用pre[]
从终点往前驱逐个输出。
变化
最短路径通常不会是唯一的,所以题目中一般会根据情景设置一些其他条件,来达到唯一路径的目的。对于这些条件首先需要观察是否满足最优子结构,如果满足,则只需要在更新最短路径时添加筛选条件即可,否则需要用vector<vector<int>> pre
来存储不同的路线,最后再使用一次dfs来筛选唯一路径。
另外,如果最后不需要输出路径,且其他条件满足最优子结构不需要dfs时,则可以不用存储pre[]
数组。