Pagini recente » Cod sursa (job #6050) | Cod sursa (job #1127447) | Cod sursa (job #2730950) | Cod sursa (job #3154334) | Cod sursa (job #406092)
Cod sursa(job #406092)
#include<stdio.h>
#include<vector>
#define INF 10000001
#define mp make_pair
#define pb push_back
#define Nmx 50005
using namespace std;
int n,m,viz[Nmx],cost[Nmx],Heap[Nmx*2],nr;
vector< pair<int,int> > G[Nmx];
void schimb(int x,int y)
{
int aux=Heap[x];
Heap[x]=Heap[y];
Heap[y]=aux;
}
void in_jos(int k)
{
while(k*2<=nr)
{
int p=k*2;
if(p+1<=nr&&cost[Heap[p+1]]<cost[Heap[p]])
++p;
if(cost[Heap[p]]>cost[Heap[k]])
return;
viz[Heap[p]]=k;
viz[Heap[k]]=p;
schimb(p,k);
k=p;
}
}
void in_sus(int k)
{
while(k>1)
{
if(cost[Heap[k/2]]<cost[Heap[k]])
return;
viz[Heap[k]]=k/2;
viz[Heap[k/2]]=k;
schimb(k,k/2);
k=k/2;
}
}
void dijkstra()
{
vector< pair<int,int> > ::iterator it;
for(int i=2;i<=n;++i)
{
viz[i]=-1;
cost[i]=INF;
}
cost[1]=0;
Heap[1]=1;
viz[1]=1;nr=1;
while(nr)
{
int min=Heap[1];
schimb(1,nr);
Heap[nr--]=0;
viz[Heap[1]]=1;
in_jos(1);
for(it=G[min].begin();it!=G[min].end();++it)
if(cost[it->first]>cost[min]+it->second)
{
cost[it->first]=cost[min]+it->second;
if(viz[it->first]==-1)
{
Heap[++nr]=it->first;
viz[it->first]=nr;
in_sus(nr);
}
else in_sus(viz[it->first]);
}
}
for(int i=2;i<=n;++i)
if(cost[i]==INF)
printf("0 ");
else printf("%d ",cost[i]);
}
void citire()
{
scanf("%d%d",&n,&m);
int x,y,c;
for(int i=1;i<=m;++i)
{
scanf("%d%d%d",&x,&y,&c);
G[x].pb(mp(y,c));
}
}
int main()
{
freopen("dijkstra.in","r",stdin);
freopen("dijkstra.out","w",stdout);
citire();
dijkstra();
return 0;
}