Pagini recente » Cod sursa (job #1893262) | Cod sursa (job #766490) | Cod sursa (job #2528738) | Cod sursa (job #737460) | Cod sursa (job #2185658)
#include <fstream>
#include <climits>
#include <vector>
using namespace std;
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
vector <int> a[50010];
vector <int> d[50010];
int n,fv[50100],w[50010],viz[50010];
struct dist
{
int poz,va;
} D[50010];
void up(int p)
{
if(p>1&&D[p].va<D[p/2].va)
{
fv[D[p].poz]=p/2;
fv[D[p/2].poz]=p;
swap(D[p],D[p/2]);
up(p/2);
}
}
void down(int p)
{
if(2*p+1<=n&&(D[p].va>D[2*p].va||D[p].va>D[2*p+1].va))
{
if(D[2*p].va<D[2*p+1].va)
{
fv[D[p].poz]=2*p;
fv[D[2*p].poz]=p;
swap(D[2*p],D[p]);
down(2*p);
}
else
{
fv[D[p].poz]=2*p+1;
fv[D[2*p+1].poz]=p;
swap(D[2*p+1],D[p]);
down(2*p+1);
}
}
else if(2*p<=n&&D[p].va>D[2*p].va)
{
fv[D[p].poz]=2*p;
fv[D[2*p].poz]=p;
swap(D[p],D[2*p]);
}
}
void dijkstra (int s)
{
int i,mini,x,k;
for(i=1; i<=n; i++)
{
D[i].va=INT_MAX/2;
D[i].poz=i;
}
D[s].va=0;
up(s);
for(k=1; k<=n; k++)
{
/* mini=INT_MAX;
x=0;
for(i=1; i<=n; i++)
if(viz[i]==0&&D[i]<mini)
{
mini=D[i];
x=i;
}
*/
mini=D[1].va;
x=D[1].poz;
// for(i=1;i<=n;i++)
// g<<D[i].va<<" "<<D[i].poz<<'\n';
// g<<'\n';
w[D[1].poz]=min(D[1].poz,mini);
if(mini!=INT_MAX/2)
{
D[1].va=INT_MAX/2;
for(i=0; i<a[x].size(); i++)
{
if(mini+d[x][i]<D[fv[a[x][i]]].va)
{
D[fv[a[x][i]]].va=mini+d[x][i];
up(fv[a[x][i]]);
}
}
down(1);
}
}
}
int m,x,y,val,i;
int main()
{
f>>n>>m;
for(i=1; i<=n; i++)
{
D[i].va=INT_MAX/2;
D[i].poz=i;
fv[i]=i;
w[i]=INT_MAX/2;
}
for(i=1; i<=m; i++)
{
f>>x>>y>>val;
a[x].push_back(y);
d[x].push_back(val);
}
dijkstra(1);
for(i=2; i<=n; i++)
if(w[i]==INT_MAX/2)g<<0<<" ";
else g<<w[i]<<" ";
return 0;
}