Cod sursa(job #3355831)

Utilizator livliviLivia Magureanu livlivi Data 26 mai 2026 19:01:12
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.98 kb
#include <fstream>
#include <vector>
#include <queue>

using namespace std;

ifstream cin("dijkstra.in");
ofstream cout("dijkstra.out");

const int INF = 1e9 + 10;

vector<int> dijkstra(int src, const vector<vector<pair<int, int>>>& edges) {
	vector<int> dist(edges.size(), INF);
	priority_queue<pair<int, int>> pq;

	dist[src] = 0;
	pq.push({0, src});

	while (!pq.empty()) {
		auto [cost, node] = pq.top();
		pq.pop();
		cost = -cost;

		if (cost != dist[node]) { continue; }

		for (auto& [nxt, c] : edges[node]) {
			if (dist[nxt] > dist[node] + c) {
				dist[nxt] = dist[node] + c;
				pq.push({-dist[nxt], nxt});
			}
		}
	}

	return dist;
}

int main() {
	int n, m; cin >> n >> m;

	vector<vector<pair<int, int>>> edges(n);
	for (int i = 0; i < m; i++) {
		int a, b, c; cin >> a >> b >> c;
		a--; b--;

		edges[a].push_back({b, c});
	}

	vector<int> dist = dijkstra(0, edges);
	for (int i = 1; i < dist.size(); i++) {
		cout << (dist[i] == INF ? 0 : dist[i]) << " ";
	}
	cout << "\n";
	return 0;
}