一个优化嘞~
注意:
1.一个点只能松弛一次(所以vis只要第一次)
2.优先队列看第一个元素排序(pair),所以不能上传0(我就因此错了好几次......)
代码:
#include<cstdio> #include<iostream> #include<queue> using namespace std; #define maxn 200005 #define INF 2147483647 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; }