Dijkstra 堆優化
阿新 • • 發佈:2018-11-28
就是個小模版qwq,提醒自己一下:
(但這是弱化版的)
#include<cstdio> #include<iostream> #include<queue> using namespace std; #define maxn 500005 #define INF 2147483647 priority_queue < pair<int,int>,vector< pair<int,int> >,greater< 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; } 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(); 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(0,e[i].to)); } } } for(int i=1; i<=n; i++) printf("%d ",dis[i]); return 0; }