Pagini recente » Cod sursa (job #822810) | Cod sursa (job #2779240) | Cod sursa (job #318670) | Cod sursa (job #2687489) | Cod sursa (job #1262269)
#include<cstdio>
#include<vector>
#define INF 2000000000
using namespace std;
vector< pair<int,int> >L[50100];
int n,m,i,j,a,b,c,poz,d[50100],v[50100],x[50100],h[50100],dh,ih;
FILE *f,*g;
void chg(int &x,int &y){
int aux=x;
x=y;
y=aux;
}
void del(){
h[1]=h[dh--];
int p=1;
int c=2;
x[h[1]]=1;
while(c<=dh){
if(d[h[c]]>d[h[c+1]]&&c<dh)
c++;
if(d[h[c]]<d[h[p]]){
chg(h[c],h[p]);
x[h[c]]=c;
x[h[p]]=p;
p=c;
c*=2;
}
else
break;
}
}
void upd(int poz){
int c=poz;
int p=poz/2;
while(p>0){
if(d[h[c]]<d[h[p]]){
chg(h[c],h[p]);
x[h[c]]=c;
x[h[p]]=p;
c=p;
p/=2;
}
else
break;
}
}
int main(){
f=fopen("dijkstra.in","r");
g=fopen("dijkstra.out","w");
fscanf(f,"%d%d",&n,&m);
for(i=1;i<=m;i++){
fscanf(f,"%d%d%d",&a,&b,&c);
L[a].push_back(make_pair(b,c));
}
for(i=2;i<=n;i++){
d[i]=INF;
}
dh=1;
h[1]=x[1]=1;
while(dh){
poz=h[1];
del();
v[poz]=1;
for(i=0;i<L[poz].size();i++){
a=L[poz][i].first;
if(d[a]>d[poz]+L[poz][i].second){
ih=(d[a]!=INF);
d[a]=d[poz]+L[poz][i].second;
if(ih==0){
dh++;
h[dh]=a;
x[a]=dh;
upd(dh);
}
else
upd(x[a]);
}
}
}
for(i=2;i<=n;i++){
if(d[i]==INF)
fprintf(g,"0 ");
else
fprintf(g,"%d ",d[i]);
}
fclose(f);
fclose(g);
return 0;
}