Cod sursa(job #2679530)

Utilizator smitoiStefan Mitoi smitoi Data 30 noiembrie 2020 18:51:03
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.13 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>

#define NMAX 100010

using namespace std;

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

int						s, t, m, n;
vector<pair<int, int>> 	listaArce[NMAX];

void dijkstra(int s) // dj astra
{
	priority_queue<pair<int, int>, vector <pair<int, int>> , greater<pair<int, int>> > q; 
	vector<int>		dist(m + 1, INT_MAX);
	vector<int>		prev(m + 1, -1);
	
    dist[s] = 0;
	q.push({0, s});
  
    dist[s] = 0;
	
	while (!q.empty())
	{
		int u = q.top().second;
		q.pop();
		
		for (int i = 0; i < listaArce[u].size(); i++)
		{
			int v = listaArce[u][i].first;
			int w = listaArce[u][i].second;
			
			if (dist[v] > dist[u] + w)
			{
				dist[v] = dist[u] + w;
				prev[v] = u;
				q.push({dist[v], v});
			}
		}
	}
	
	for (int i = 2; i <= n; i++)
		g << dist[i] << ' ';
}

int		main()
{
	f >> n >> m;
	for (int i = 0; i < m; i++) // m complexitatea pentru citire
	{
		int a, b, w;
		f >> a >> b >> w;
		listaArce[a].push_back({b, w});
	}
	
	dijkstra(1); // facem djkstra unde vom afisa si drumul minim
	
	return 0;
}