Dijkstra堆优化算法的内容大致如下:
首先开一个堆,用dist数组存储起点到每个点的最小距离,初始化dist数组为最大值。
接着从起点出发,将当前点到其他能一步到达的点之间的距离进行比较和更新,那么得到的最小的那个距离就可以确定为当前点到这个点的最小距离,并且可以保证不会再有更小的距离。于是我们将每个距离加入堆中,并在下一次取出堆顶元素作为“起点”,重复上述步骤。
同时,我们需要使用vis数组来存储点,并将已经作为过“起点”的点进行标记。
最后,当堆为空时,所有点到起点的最小值便已知。
题目描述
给定一个 n 个点,m 条有向边的带非负权图,请你计算从 s 出发,到每个点的距离。
数据保证你能从 s 出发到任意点。
输入格式
第一行为三个正整数n,m,s。 第二行起 m 行,每行三个非负整数ui,vi,wi,表示从 ui 到 vi 有一条权值为 wi 的有向边。
输出格式
输出一行 n 个空格分隔的非负整数,表示 s 到每个点的距离
下面用Dijkstra堆优化算法来解决此题。
代码如下:
#include<bits/stdc++.h>
using namespace std;
int n,m,s;
struct node{
int u,nxt;
long long w;
}edge[200010];
int tot=0;
int head[100010];
void add(int x,int y,int z){
tot++;
edge[tot].nxt=head[x];
edge[tot].u=y;
edge[tot].w=z;
head[x]=tot;
} //用链式前向星存储图
priority_queue<pair<long long,int> > q;//先开大根堆
long long dist[100010];
bool vis[100010];
void dijkstra(){
memset(dist,0x3f,sizeof(dist));
dist[s]=0;
q.push(make_pair(0,s));
while(q.size()){
int topp=q.top().second;
q.pop();
if(vis[topp])continue;
vis[topp]=1;
for(int i=head[topp];i;i=edge[i].nxt){
int v=edge[i].u;
if(dist[v]>dist[topp]+edge[i].w){
dist[v]=dist[topp]+edge[i].w;
q.push(make_pair(-dist[v],v));//更新dist的值为最小,并存入堆,但为了让堆顶为最小值,在dist[v]前加上负号
}
}
}
}
int main(){
scanf("%d%d%d",&n,&m,&s);
int x,y,z;
for(int i=1;i<=m;i++){
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);//存储图
}
dijkstra();
for(int i=1;i<=n;i++){
printf("%lld ",dist[i]);
}
return 0;
}