由于用优先队列的Dijkstra在队列中一个点可能出现很多次,所以复杂度为 O(n+m)log(n+m),但是pd_ds优化的Dijkstra,复杂度可以达到 O(n+m)log(n)。
注:pb_ds中不会自动去重。
代码:
#include<cstdio> #include<iostream> #include<queue> #include<ext/pb_ds/priority_queue.hpp> using namespace std; #define maxn 200005 #define INF 2147483647 __gnu_pbds ::priority_queue< pair<int,int> > q; struct node { int nxt,to,w; } e[maxn]; int dis[maxn],head[maxn],cnt,n,m,s; void add(int a,int b,int c) { e[++cnt].to=b; e[cnt].w=c; e[cnt].nxt=head[a]; head[a]=cnt; } bool vis[maxn]; int main() { scanf("%d%d%d",&n,&m,&s); for(int i=1; i<=n; i++) dis[i]=INF; dis[s]=0; for(int i=1; i<=m; i++) { int a,b,c; scanf("%d%d%d",&a,&b,&c); add(a,b,c); } q.push(make_pair(0,s)); while(!q.empty()) { int p=q.top().second; q.pop(); if(vis[p]) continue; vis[p] = 1; for(int i=head[p]; i; i=e[i].nxt) { if(dis[p]!=INF&&dis[e[i].to]>dis[p]+e[i].w) { dis[e[i].to]=dis[p]+e[i].w; q.push(make_pair(-dis[e[i].to],e[i].to)); } } } for(int i=1; i<=n; i++) printf("%d ",dis[i]); return 0; }