Cod sursa(job #357120)

Utilizator Mishu91Andrei Misarca Mishu91 Data 17 octombrie 2009 23:07:40
Problema Algoritmul lui Dijkstra Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.13 kb
#include <cstdio>
#include <vector>
#include <bitset>
#include <queue>

using namespace std;

#define MAX_N 50005
#define INF 0x3f3f3f3f

#define foreach(V) for(typeof (V).begin() it = (V).begin(); it != (V).end(); ++it)
#define x first
#define c second

vector < pair<int, int> > G[MAX_N];
int D[MAX_N];
int N, M;


struct cmp
{
	bool operator()(const int &a, const int &b)
	{
		return D[a] > D[b];
	}
};

priority_queue <int, vector <int>, cmp> Q;

void citire()
{
	scanf("%d %d",&N, &M);
	
	while(M--)
	{
		int x, y, c;
		scanf("%d %d %d",&x, &y, &c);
		G[x].push_back( make_pair(y,c) );
	}
}

void solve()
{
	memset(D, INF, sizeof D);
	
	bitset <MAX_N> inq;
	inq[1] = 1;
	D[1] = 0;
	
	for(Q.push(1);!Q.empty(); Q.pop())
	{
		int nod = Q.top();
	//	inq[nod] = 0;
		
		foreach(G[nod])
			if(D[it -> x] > D[nod] + it -> c)
			{
				D[it -> x] = D[nod] + it -> c;
				if(!inq[it -> x])
				{
					Q.push(it -> x);
					inq[it -> x] = 1;
				}
			}
	}
	
	for(int i = 2; i <= N; ++i)
		if(D[i] == INF) printf("0 ");
		else printf("%d ",D[i]);
}

int main()
{
	freopen("dijkstra.in","rt",stdin);
	freopen("dijkstra.out","wt",stdout);
	
	citire();
	solve();
}