Cod sursa(job #937240)

Utilizator xbogdanBogdan Boamfa xbogdan Data 9 aprilie 2013 23:47:17
Problema Algoritmul lui Dijkstra Scor 80
Compilator cpp Status done
Runda Arhiva educationala Marime 1.36 kb
#include "iostream"
#include "fstream"
#include "queue"
#include "vector"

#define pr pair<int,int>
//#define INF (1 << 30)

using namespace std;

const int NMAX = 50006;
const int INF = 20000;

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

struct Compare 
{
	bool operator() (const pr &a, const pr &b) const
	{	
		return a.second < b.second;
	}
};

priority_queue < pr, vector< pr >, Compare> Q;
vector < pr > *Graf;
int nr_nod;
int* dist;

void Read()
{
	int from,to,by,muchii;
	in >> nr_nod >> muchii;
	dist = new int[nr_nod+1];
	Graf = new vector<pr>[nr_nod+1];
	for (int i = 0; i < muchii; ++i)
	{
		in >> from >> to >> by;
		Graf[from].push_back(make_pair(to,by));
	}	
}
void Dijkstra(int sursa)
{
	int to,by,from;
	for (int i = 1; i < nr_nod+1; ++i)
		dist[i] = INF;
	dist[sursa] = 0;
	Q.push(pr(sursa,dist[sursa]));
	
	while (!Q.empty())
	{
		from = Q.top().first;
		Q.pop();
		int size = Graf[from].size();
		for (int i = 0; i < size; ++i)
		{
			to = Graf[from][i].first;
			by = Graf[from][i].second;
			//cout<<from<<" "<<to<<" "<<by<<"\n";
			if (dist[to] > dist[from] + by)
			{
				dist[to] = dist[from] + by;
				Q.push(make_pair(to,dist[to]));
			}
		}
	}
}
void Print()
{
	for (int i = 2; i < nr_nod+1; ++i)
		if(dist[i] == INF)
			out << "0 ";
		else
			out << dist[i] << " ";
}
int main(int argc, char const *argv[])
{
	Read();
	Dijkstra(1);
	Print();
	return 0;
}