Pagini recente » Cod sursa (job #1041917) | Cod sursa (job #2363235) | Cod sursa (job #271801) | Cod sursa (job #1745588) | Cod sursa (job #2185692)
#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)
{
swap(D[p],D[p/2]);
fv[D[p].poz]=p;
fv[D[p/2].poz]=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)
{
swap(D[2*p],D[p]);
fv[D[p].poz]=p;
fv[D[2*p].poz]=2*p;
down(2*p);
}
else
{
swap(D[2*p+1],D[p]);
fv[D[p].poz]=p;
fv[D[2*p+1].poz]=2*p+1;
down(2*p+1);
}
}
else if(2*p<=n&&D[p].va>D[2*p].va)
{
swap(D[p],D[2*p]);
fv[D[p].poz]=p;
fv[D[2*p].poz]=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;
}
w[s]=0;
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]=D[1].poz;
if(mini!=INT_MAX/2)
{
D[1].va=INT_MAX/2;
down(1);
for(i=0; i<a[x].size(); i++)
{
if(mini+d[x][i]<w[a[x][i]])
{
w[a[x][i]]=mini+d[x][i];
D[fv[a[x][i]]].va=mini+d[x][i];
up(fv[a[x][i]]);
}
}
}
}
}
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(2);
for(i=2; i<=n; i++)
if(w[i]==INT_MAX/2)g<<0<<" ";
else g<<w[i]<<" ";
return 0;
}