Pagini recente » Cod sursa (job #1351344) | Cod sursa (job #281090) | Cod sursa (job #1520160) | Cod sursa (job #2025909) | Cod sursa (job #2162968)
#include <fstream>
#include <vector>
using namespace std;
ifstream cin ("dijkstra.in");
ofstream cout("dijkstra.out");
const int INF = 999999999;
const int N = 50005;
vector <int> a[N];
vector <int> c[N];
int x, y, cost, n, m, h[N], poz[N], d[N], p, q, nh;
void citire()
{
int x,y,cost;
cin>>n>>m;
for(int i=0; i<=m; i++)
{
cin>>x>>y>>cost;
a[x].push_back(y);
c[x].push_back(cost);
}
}
void swapp(int p, int q)
{
swap(h[p], h[q]);
poz[h[p]]=p;
poz[h[q]]=q;
}
void urca(int p)
{
while(p>1 && d[h[p]]<d[h[p/2]])
{
swapp(p, p/2);
p/=2;
}
}
void coboara(int p)
{
int fs=2*p, fd=2*p+1, bun=p;
if(fs<=nh && d[h[fs]]<d[h[bun]])
{
bun=fs;
}
if(fd<=nh && d[h[fd]]<d[h[bun]])
{
bun=fd;
}
if(bun!=p)
{
swapp(p, bun);
coboara(bun);
}
}
void adauga(int p)
{
h[++nh]=p;
poz[p]=nh;
urca(nh);
}
void sterge(int p)
{
swapp(p, nh--);
urca(p);
coboara(p);
}
void dijkstra(int x0)
{
nh=n;
for(int i=1; i<=n; i++)
{
d[i]=INF;
h[i]=i;
poz[i]=i;
}
d[x0]=0;
urca(poz[x0]);
while(nh>0 && d[h[1]] != INF)
{
int x=h[1];
sterge(1);
for(int i=0; i<a[x].size(); i++)
{
int y=a[x][i];
cost=c[x][i];
if(d[x]+cost<d[y])
{
d[y]=d[x]+cost;
//pred[y]=x;
urca(poz[y]);
}
}
}
}
int main()
{
citire();
dijkstra(1);
for(int i=2; i<=n; i++)
cout<<d[i]<<' ';
return 0;
}