Cod sursa(job #399483)

Utilizator MESAROSLaura Mesaros MESAROS Data 20 februarie 2010 15:35:45
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.14 kb
#include <fstream>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;

const int MAXN = 50005;
const int INF = 1000000;

int N, M;
vector< pair<int, int> > G[MAXN];
int Dmin[MAXN];

void citeste() {
	ifstream fin("dijkstra.in");

	fin >> N >> M;
	for (int i = 0; i < M; ++i) {
		int a, b, c;

		fin >> a >> b >> c;
		G[a].push_back(make_pair(b, c));
	}
}

void dijkstra() {
	bool viz[MAXN];
	queue<int> Q;

	memset(Dmin, INF, sizeof(Dmin));
	memset(viz, false, sizeof(viz));
	cout<<"/n";


	Dmin[1] = 0;
	Q.push(1);
	viz[1] = true;

	while (!Q.empty()) {
		int nod = Q.front();
		Q.pop();
		viz[nod] = false;

		for (vector< pair<int, int> >::iterator it = G[nod].begin(); it != G[nod].end(); ++it)
			if (Dmin[nod] + it->second < Dmin[it->first]) {
				Dmin[it->first] = Dmin[nod] + it->second;
				if (!viz[it->first]) {
					Q.push(it->first);
					viz[it->first] = true;
				}
			}
	}
}

void afisare() {
	ofstream fout("dijkstra.out");

	for (int i = 2; i <= N; ++i)
		fout << (Dmin[i] < INF ? Dmin[i] : 0) << " ";
}

int main() {
	citeste();
	dijkstra();
	afisare();
}