Cod sursa(job #663260)

Utilizator alexdmotocMotoc Alexandru alexdmotoc Data 18 ianuarie 2012 09:53:39
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.36 kb
#include <iostream>
#include <cstdio>
#include <vector>

using namespace std;

#define PB push_back
#define MKP make_pair
#define maxN 500005
#define INF 0x3f3f3f3f

vector < pair <int , int> > lista[maxN];

int cost[maxN] , N , M , sol[maxN];
bool viz[maxN];

int main ()
{
	freopen ("dijkstra.in" , "r" , stdin);
	freopen ("dijkstra.out" , "w" , stdout);
	
	scanf ("%d %d" , &N , &M);
	
	int a , b , c;
	
	for (int i = 1 ; i <= M ; ++i)
	{
		scanf ("%d %d %d" , &a , &b , &c);
		
		lista[a]. PB (MKP (b , c));
	}
	
	for (int i = 1 ; i <= N ; ++i)
		cost[i] = INF;
	
	cost[1] = 0;
	
	int nod , dim = 0 , cost_n;
	int nodcur , costcur;
	
	while (dim < N)
	{
		cost_n = INF;
		
		for (int i = 1 ; i <= N ; ++i)
		{
			if (viz[i])
				continue;
			
			if (cost[i] < cost_n)
			{
				cost_n = cost[i];
				nod = i;
			}
		}
		
		viz[nod] = true;
		sol[nod] = cost_n;
		++dim;

		
		for (unsigned i = 0 ; i < lista[nod]. size () ; ++i)
		{
			nodcur = lista[nod][i].first;
			costcur = lista[nod][i].second;
			
			if (viz[nodcur])
				continue;
			
			if (cost[nod] + costcur >= cost[nodcur])
				continue;
			
			cost[nodcur] = cost[nod] + costcur;
			
		}
		
	}
	
	for (int i = 2 ; i <= N ; ++i)
	{
		if (sol[i] == INF)
		{
			printf ("0 ");
			continue;
		}
		
		printf ("%d " , sol[i]);
	}
	
	return 0;
}