Pagini recente » Cod sursa (job #660343) | Cod sursa (job #1876642) | Cod sursa (job #2537828) | Cod sursa (job #1464791) | Cod sursa (job #2208487)
#include <fstream>
#include <vector>
#include <set>
using namespace std;
set <pair <int , int> >heap;
vector <int> arc[50002] , cost[50002];
int n,m,i,a,b,c,dis[50002],nod,Min;
int main()
{
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
f>>n>>m;
for(i=1;i<=m;i++)
{
f>>a>>b>>c;
arc[a].push_back(b);
cost[a].push_back(c);
}
for(i=2;i<=n;i++)
dis[i]=2000000000;
for(i=0;i<arc[1].size();i++)
{
dis[arc[1][i]]=cost[1][i];
heap.insert(make_pair(cost[1][i],arc[1][i]));
}
while(!heap.empty())
{
nod=heap.begin()->second;
Min=heap.begin()->first;
heap.erase(heap.begin());
for(i=0;i<arc[nod].size();i++)
if(dis[arc[nod][i]] > Min+cost[nod][i])
{
if(dis[arc[nod][i]]==2000000000)
heap.erase(make_pair(dis[arc[nod][i]],arc[nod][i]));
dis[arc[nod][i]] = Min+cost[nod][i];
heap.insert(make_pair(dis[arc[nod][i]],arc[nod][i]));
}
}
for(i=2;i<=n;i++)
if(dis[i]==2000000000)
g<<0<<' ';
else
g<<dis[i]<<' ';
return 0;
}