Cod sursa(job #1517061)

Utilizator adimAlexander Dmitriev adim Data 3 noiembrie 2015 20:25:36
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.24 kb
#include <bits/stdc++.h>
#define inf 1<<30
#define NMax 50005

#define mp make_pair
#define pb push_back
using namespace std;

ifstream f("dijkstra.in");
ofstream g("dijkstra.out");

struct cmp {
	bool operator()(const pair<int, int> &a, const pair<int, int> &b) {
		if (a.first < b.first)
			return false;
		if (a.first == b.first && a.second < b.second)
			return false;
		return true;
	}
};

priority_queue<pair<int, int>, vector< pair<int, int> >, cmp> q;
int n, m;
vector<int> V[NMax];
vector<int> C[NMax];
int R[NMax];

void read() {
	f>>n>>m;
	R[1] = 0;
	for (int i=2;i<=n;i++)
		R[i] = inf;

	for (int i=1;i<=m;i++) {
		int a,b,c;
		f>>a>>b>>c;
		V[a].pb(b);
		C[a].pb(c);
	}
}

void dijkstra() {
	q.push(mp(0,1));
	
	while (!q.empty()) {
		int nod = q.top().second;
		int cost = q.top().first;
		q.pop();
		
		for (unsigned i=0;i<V[nod].size();i++) {
			int vecin = V[nod][i];
			int muchie = C[nod][i];
			
			if (R[vecin] > muchie + cost) {
				R[vecin] = muchie+cost;
				q.push(mp(R[vecin], vecin));
			}
		}
	}
}

void output() {
	for (int i=2;i<n;i++) {
		if (R[i] == inf)
			g<<0<<' ';
		else
			g<<R[i]<<' ';
	}
	
	if (R[n] == inf)
		g<<0<<'\n';
	else
		g<<R[n]<<'\n';
}

int main() {

	read();
	dijkstra();
	output();
	
	f.close(); g.close();
	return 0;
}