Cod sursa(job #1126218)

Utilizator httpsLup Vasile https Data 26 februarie 2014 22:01:26
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.05 kb
#include <iostream>
#include <cstring>
#include <fstream>
#include <set>
#include <vector>
using namespace std;

ifstream f("dijkstra.in");
ofstream g("dijkstra.out");

#define cout g
#define nmax 50010
#define inf 0x3f3f3f3f

vector <pair<int,int> > vecini[nmax];
int n,m,a,b,c,i;
int distanta[nmax];
set <pair<int,int> > heap;
int nod,cost;

int main()
{
f>>n>>m;
memset(distanta,inf,(n+1)*sizeof(int));
for(;m;--m)
{
    f>>a>>b>>c;
    vecini[a].push_back(make_pair(b,c));
    vecini[b].push_back(make_pair(a,c));
}
distanta[1]=0;
heap.insert(make_pair(0,1));
while (!heap.empty())
{
    cost=heap.begin()->first;
    nod=heap.begin()->second;
    heap.erase(heap.begin());
    for(i=0;i<vecini[nod].size();++i)
    if(distanta[vecini[nod][i].first]>distanta[nod]+vecini[nod][i].second)
    {
        distanta[vecini[nod][i].first]=distanta[nod]+vecini[nod][i].second;
        heap.insert(make_pair(distanta[vecini[nod][i].first],vecini[nod][i].first));
    }
}
 for(i=2;i<=n;++i) cout<<distanta[i]<<' ';
    return 0;
}