Cod sursa(job #1449822)

Utilizator cristian.caldareaCaldarea Cristian Daniel cristian.caldarea Data 10 iunie 2015 18:40:51
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.15 kb
#include <cstdio>
#include <cstring>
#include <set>
#include <vector>
using namespace std;
 
#define pb push_back
#define mp make_pair
#define MAXN 50100
#define INF 1000000000
 
int N, M, d[MAXN]; 
vector<vector<pair<int, int> > > G;
typedef vector<pair<int, int> >::iterator IT;
 
set<pair<int, int> > T;
 
void Solve(void)
{
    int i, x;
     
    for(i = 2; i <= N; i++) d[i] = INF;
    T.insert( mp(0, 1) );
	int y, dx, cost;
    while ( !T.empty() )
    {
        dx = T.begin()->first, x = T.begin()->second;
        T.erase(T.begin());
        if ( dx > d[x] )
			continue;
        for(auto p : G[x] )
        {
			y = p.first;
			cost = p.second; // muchia
			if(d[y] > dx + cost )   // if ( d[y] > d[x] + cost )
			{
				d[y] = dx + cost, 
				T.insert(mp(d[y], y));  
			}
		}
    }
}
 
int main(void)
{
    freopen("dijkstra.in", "rt", stdin);
    freopen("dijkstra.out", "wt", stdout);
 
    int i, a, b, c;
 
    scanf("%d %d\n", &N, &M);
	G.resize(N + 1);
    for(i = 1; i <= M; i++)
        scanf("%d %d %d\n", &a, &b, &c), 
        G[a].pb(mp(b, c));
 
    Solve();
 
    for(i = 2; i <= N; i++)
        printf("%d ", d[i] == INF ? 0 : d[i]);
 
    return 0;
}