Cod sursa(job #402485)

Utilizator Bogdan_CCebere Bogdan Bogdan_C Data 23 februarie 2010 21:31:33
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.05 kb
#include<fstream>
#include<vector>
#include<queue>
using namespace std;
#define nmax 50005
#define pb push_back
#define inf 1<<30
vector<int> g[nmax],c[nmax];
int viz[nmax],d[nmax];
int n,m,s=1;
void citire ()
{
	freopen("dijkstra.in","r",stdin);
	int i,a,b,cost;
	scanf("%d %d",&n,&m);
	for(i=1;i<=m;i++)
	{
	     scanf("%d %d %d",&a,&b,&cost);
		g[a].pb(b);
		c[a].pb(cost);
	}

}
struct comp
{
	bool operator()(int i,int j)
	{
		return d[i]>d[j];
	}
};
priority_queue<int,vector<int>,comp> q;
void dijkstra(int nod)
{
	int i,k;
	for(i=1;i<=n;i++)
		d[i]=inf;
	q.push(nod);
	d[nod]=0;
	while(!q.empty())
	{
		k=q.top();
		viz[k]=0;
		q.pop();
		for(i=0;i<g[k].size();i++)
			if(d[g[k][i]]>d[k]+c[k][i])
			{
				d[g[k][i]]=d[k]+c[k][i];
				if(viz[g[k][i]]!=1)
					q.push(g[k][i]),viz[g[k][i]]=1;
			}
	}
}
void scrie()
{
	int i;
	freopen("dijkstra.out","w",stdin);
	for(i=2;i<=n;i++)
		if(d[i]==inf) printf("0 ");
		else printf("%d ",d[i]);
	
}
int main ()
{
	citire();
	dijkstra(s);
	scrie();
	return 0;
}