dijkstra是图论中解决最短路问题的重要算法之一,必须熟练掌握
dijkstra用于解决单源最短路问题,要求图中必须不存在负边权
朴素dijkstra的时间复杂度为O(n^2),堆优化版dijkstra的时间复杂度为O(mlogn),两者的主要差异在于堆优化版采用优先队列来存储当前所在的点和与这个点联通的点之间的距离,从而免去了一次寻找与当前点联通且距离最小的点的循环
下面是代码,不易理解的地方会有注释,帮助理解
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#define inf 0x3f3f3f3f//无穷大
using namespace std;
const int N=1e6;
int e[N*2],ne[N*2],h[N],cnt=0,w[N],dis[N],st[N]={0},s,m,n;
//w用于存储边的权值;dis存储每个点到源点的距离;st为状态数组,1表示当前点已找到最短路,0则相反
//s为初始点,m为边的数量,n为点的数量
typedef pair<int,int> pii;//first为到源点距离,second为节点编号
priority_queue<pii,vector<pii>,greater<pii> >heap;//定义优先队列,注意greater后一定加一个空格
//stl的优先队列默认为大根堆,加上greater后为小根堆
void add(int a,int b,int c)//采取邻接表存储图
{
e[cnt]=b;
ne[cnt]=h[a];
w[cnt]=c;
h[a]=cnt++;