Cod sursa(job #401668)

Utilizator cezyGrigore Cezar cezy Data 22 februarie 2010 23:58:31
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 0.91 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];
struct cmp 

{ 

bool operator() (const int &a, const int &b) const

{ 

return d[a]>d[b]; 

} 

}; priority_queue <int, vector <int>,cmp> q2;
int viz[nmax];
int n,sf,inc,m;
int main ()
{
	ifstream f("dijkstra.in");
    ofstream fout("dijkstra.out");
	int i,j,k,a,b,t;
	f>>n>>m;
	for(i=1;i<=m;i++)
		f>>a>>b>>t,g[a].pb(b),c[a].pb(t);
	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++;
	}
	for(i = 2; i <= n; i++)
        if(d[i]==inf) fout<<0<<' ';
		else fout<<d[i]<<' ';
	f.close();
	return 0;
}