Pagini recente » Cod sursa (job #748813) | Cod sursa (job #2860802)
#include <iostream>
#include <fstream>
#include <algorithm>
#include <limits.h>
using namespace std;
/// infoarena dijkstra
ifstream fin("dijkstra.in");ofstream fout("dijkstra.out");
int n,m,p,id[50002],v[50001];
///n-nr noduri;m-nr muchii;p-sursa
struct muchie
{
int x,y,d;
}U[250001];
bool ordine(muchie id1,muchie id2)
{
if (id1.x!=id2.x)
return id1.x<id2.x;
if (id1.y!=id2.y)
return id1.y<id2.y;
return id1.d<id2.d;
}
int find_id(int val,int lo,int hi)
{
if(lo>hi)
return -1;
else
{
int mid=(lo+hi)/2;
if (U[mid].x==val)
{
while (U[mid].x==val)
mid--;
return mid+1;
}
else
{
if (U[mid].x>val)
return find_id(val,lo,mid-1);
else
return find_id(val,mid+1,hi);
}
}
}
void Dijkstra()
{
int f[n+1];
for (int i=1;i<=n;i++)
f[i]=0;
v[0]=INT_MAX;v[1]=0;f[1]=1;
for (int i=id[1];U[i].x==U[id[1]].x;i++)
v[U[i].y]=U[i].d;
for (int k=1;k<=n-1;k++)
{
int imin=0;
for (int i=1;i<=n;i++)
if (v[i]<v[imin] && f[i]==0)
imin=i;
f[imin]=1;
if (imin!=0 && id[imin]!=-1)
{
for (int i=id[imin];U[i].x==U[id[imin]].x;i++)
{
if (f[U[i].y]==0 && v[U[i].y]>v[U[i].x]+U[i].d)
v[U[i].y]=v[U[i].x]+U[i].d;
}
}
}
}
int main()
{
fin>>n>>m;
for (int i=1;i<=m;i++)
fin>>U[i].x>>U[i].y>>U[i].d;
sort(U+1,U+m+1,ordine);
id[0]=0;id[n+1]=m+1;
for (int i=1;i<=n;i++)
{
v[i]=INT_MAX;
id[i]=find_id(i,id[i-1]+1,m+1);
}
Dijkstra();
for (int i=2;i<=n;i++)
{
if (v[i]!=INT_MAX)
fout<<v[i]<<" ";
else
fout<<0<<" ";
}
return 0;
}