Cod sursa(job #663328)

Utilizator alexdmotocMotoc Alexandru alexdmotoc Data 18 ianuarie 2012 11:57:01
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.99 kb
#include <iostream>
#include <cstdio>
#include <vector>

using namespace std;

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

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

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

int pozH[maxN] , H[maxN] , dim;

void push (int nod)
{
	int tata = nod / 2;
	
	if (nod == 1) return;
	
	if (cost[H[tata]] <= cost[H[nod]])
		return;
	
	swap (H[tata] , H[nod]);
	swap (pozH[H[tata]] , pozH[H[nod]]);
	
	push (tata);
}



void pop (int nod)
{
	int f1 , f2 , nodmin = nod;
	
	f1 = nod * 2;
	f2 = nod * 2 + 1;
	
	if (cost[H[f1]] < cost[H[nodmin]] && f1 <= dim)
		nodmin = f1;
	
	if (cost[H[f2]] < cost[H[nodmin]] && f2 <= dim)
		nodmin = f2;
	
	if (nodmin == nod)
		return;
	
	swap (H[nod] , H[nodmin]);
	swap (pozH[H[nod]] , pozH[H[nodmin]]);
	
	pop (nodmin);

}




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 <= N ; ++i)
		cost[i] = INF;
	
	cost[1] = 0;
	
	for (int i = 1 ; i <= N ; ++i)
	{
		pozH[i] = ++dim;
		H[dim] = i;
		
		push (pozH[i]);
	}
	
	for (int i = 1 ; i <= M ; ++i)
	{
		scanf ("%d %d %d" , &a , &b , &c);
		
		lista[a]. PB (MKP (b , c));
	}
	
	int nod , nodcur , costmuc;
	
	while (dim)
	{
		nod = H[1];
		
		viz[nod] = true;
		
		swap (H[1] , H[dim]);
		swap (pozH[H[1]] , pozH[H[dim]]);
		
		--dim;
		
		pop (pozH[1]);
		
		for (unsigned i = 0 ; i < lista[nod]. size () ; ++i)
		{
			nodcur = lista[nod][i].first;
			costmuc = lista[nod][i].second;
			
			if (viz[nodcur])
				continue;
			
			if (cost[nod] + costmuc < cost[nodcur])
			{
				cost[nodcur] = cost[nod] + costmuc;
				
				push (pozH[nodcur]);
				pop (pozH[nodcur]);
			}
		}
	}
	
	for (int i = 2 ; i <= N ; ++i)
	{
		if (cost[i] == INF)
		{
			printf ("0 ");
			continue;
		}
		
		printf ("%d " , cost[i]);
	} 
	
	return 0;
}