Cod sursa(job #556487)

Utilizator cezyGrigore Cezar cezy Data 16 martie 2011 10:08:58
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 0.87 kb
#include<fstream>
#include<vector>
#include<queue>
using namespace std;
#define nmax 50100
#define pb push_back
#define inf 1000000000
vector<int> g[nmax],c[nmax];
int d[nmax],q[1<<20];
priority_queue <int> q2;
int viz[nmax];
int n,sf,inc,m;
void citire()
{
	ifstream fin("dijkstra.in");
	fin>>n>>m;
	int i,a,b,dist;
	for(i=1;i<=m;i++)
		fin>>a>>b>>dist,g[a].pb(b),c[a].pb(dist);
	fin.close();
}
void scrie()
{
	ofstream fout("dijkstra.out");
	int i;
	for(i = 2; i <= n; i++)
        if(d[i]==inf) fout<<0;
		else fout<<d[i]<<' ';
	fout.close();
}
int main ()
{
	citire();
	int i,j,k,a;
    for(i=1;i<=n;i++)
		d[i]=inf;
	inc=sf=0;
	q2.push(1);d[1]=0;
	while(inc<=sf)
	{
		a=q2.top();
		q2.pop();inc++;
		for(i = 0; i < g[a].size(); i++)
        if( d[ g[a][i] ] > d[a] + c[a][i] )
          d[ g[a][i] ] = d[a] + c[a][i], q2.push( g[a][i]),sf++;
	}
	scrie();
	return 0;
}