Cod sursa(job #1042952)

Utilizator bodyionitaIonita Bogdan Constantin bodyionita Data 27 noiembrie 2013 20:22:38
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.08 kb
#include <cstdio>
#include <set>
#include <vector>
using namespace std;
#define INF 100000000
#define NMAX 200001
int D[NMAX], N, M, i;
vector< pair<int, int> > G[NMAX];
inline void solve()
{
    set< pair<int, int> > H;
    vector< pair<int, int> >::iterator it;
    int cost, nod,i;
    for (i=2; i<=N; i++) D[i]=INF;
    H.insert( make_pair(0, 1) );
    while ( H.size()>0 )
    {
        cost=( *H.begin() ).first;
        nod=( *H.begin() ).second;
        H.erase( *H.begin() );
        for (it=G[nod].begin(); it!=G[nod].end(); it++)
            if ( D[(*it).second]> cost+ (*it).first )
                D[(*it).second]= cost+ (*it).first, H.insert( make_pair(D[(*it).second], (*it).second) );
    }
}
int main()
{
    freopen("dijkstra.in", "r", stdin);
    freopen("dijkstra.out", "w", stdout);
    scanf("%d%d", &N, &M);
    for (i=1; i<=M; i++)
    {
        int x, y, c;
        scanf("%d%d%d", &x, &y, &c);
        G[x].push_back( make_pair(c, y) );
    }
    solve();
    for (i=2; i<=N; i++)
        printf("%d ", D[i]==INF ? 0 : D[i]);
    return 0;
}