背景
作为一个只会spfa求最短路的蒟蒻,我挂在了[NOI2018]归程这道题上.
是我了。。。
所以,我决定学习dijkstra算法.
实际上也蛮简单的嘛。。。
算法描述
从起点每次新拓展一个到起点距离最短的点,再以这个点为中间点,更新起点到其他所有点的距离.
算法正确性证明:
由于所有边权都为正,故不会存在一个距离更短的没被扩展过的点,所以这个点的距离永远不会再被改变
算法时间复杂度为 O ( n 2 ) O(n^2) O(n2),因为每次查找新拓展的距离最短的点需耗时 O ( n ) O(n) O(n),所以用堆优化使得时间复杂度降为 O ( ( n + m ) l o g   m ) O((n+m)log\:m) O((n+m)logm).
程序实现
#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
#define maxn 200010
using namespace std;
struct edge{
int v,w,next;
}e[maxn];
int n,m,s,dis[maxn];
bool vis[maxn];
int head[maxn],cnt;
void add(int u,int v,int w){
e[++cnt].v =v;
e[cnt].w =w;
e[cnt].next =head[u];
head[u]=cnt;
}
struct node{
int pos;
int dis;
bool operator <(node nd)const {
return nd.dis <dis;//优先队列默认大根堆,重载为小根堆(重定义相反即可)
//return dis>nd.dis ;//标准形式,可以发现小于号重载为了大于号
}
};
priority_queue<node>q;
void dijkstra(){
dis[s]=0;
q.push((node){s,0});
while(!q.empty()){
node op=q.top();
q.pop();
int u=op.pos,d=op.dis ;
if(vis[u])continue;//这东西是为了防止有一些点由于路径在 已经进入prq但是还没有被操作时 又被更新,导致prq里面不止一个u,第二次遇到就不要再用它更新,跳过好了
vis[u]=true;//一个已经操作过的点绝对不会再次被操作
for(int i=head[u];i;i=e[i].next ){
int v=e[i].v ;
if(dis[v]>dis[u]+e[i].w ){
dis[v]=dis[u]+e[i].w ;
if(!vis[v])q.push((node){v,dis[v]});
}
}
}
}
int main(){
scanf("%d%d%d",&n,&m,&s);
for(int i=1;i<=n;i++)dis[i]=inf;
for(int i=1,u,v,w;i<=m;i++){
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
}
dijkstra();
for(int i=1;i<=n;i++)printf("%d ",dis[i]);
printf("\n");
return 0;
}