【模板】单源最短路径*

 1 #include<cstdio>
 2 #define LL long long
 3 #define maxn 10010
 4 #define maxm 500010
 5 #define inf 2147483647
 6 int n,m,q,w[maxn],in[maxn];
 7 int a,b,c,l,r;
 8 int p[maxn],ps,top;
 9 bool v[maxn];
10 struct edge{int s,n,v;}e[maxm];
11 struct heap{int s,v;}h[maxm];
12 inline void add(int x,int y,int z){e[++ps]=(edge){y,p[x],z},p[x]=ps;}
13 inline LL min(LL x,LL y,LL z){return x<y+z?x:y+z;}
14 void swap(int x,int y,int xx,int yy){
15     h[0]=(heap){h[x].s,h[x].v};
16     h[x]=(heap){h[y].s,h[y].v};
17     h[y]=(heap){h[0].s,h[0].v};
18     in[xx]=y;in[yy]=x;
19 }
20 void push(int x,int y,int z){
21     h[z]=(heap){x,y},in[x]=z;
22     while(in[x]>1){
23         if(h[in[x]].v<h[in[x]>>1].v) swap(in[x],in[x]>>1,x,h[in[x]>>1].s);
24         else break;
25     }
26 }
27 int pop(){
28     int ret=h[1].s;
29     h[1]=(heap){h[top].s,h[top--].v};
30     int x=h[1].s,z;
31     in[h[1].s]=1;
32     while(in[x]<=top){
33         l=in[x]<<1,r=l+1;
34         if(h[l].v<h[r].v) z=l;
35         else z=r;
36         if(h[z].v<h[in[x]].v) swap(in[x],z,x,h[z].s);
37         else break;
38     }
39     return ret;
40 }
41 int main(){
42     scanf("%d%d%d",&n,&m,&q);
43     for(int i=1;i<=n;i++) if(i!=q) w[i]=inf;
44     for(int i=1;i<=m;i++){
45         scanf("%d%d%d",&a,&b,&c);
46         add(a,b,c);
47     }
48     for(int k=0;k<n;k++){
49         if(k) q=pop();
50         v[q]=1;
51         for(int i=p[q];i;i=e[i].n){
52             a=e[i].s,b=min(w[a],w[q],e[i].v);
53             if(b<w[a]){
54                 w[a]=b;
55                 if(in[a]) push(a,w[a],in[a]);
56                 else push(a,w[a],++top);
57             }
58         }
59     }
60     for(int i=1;i<=n;i++) printf("%d ",w[i]);
61     return 0;
62 }