Cod sursa(job #381783)

Utilizator lorandCsorba Lorand-Alexandru lorand Data 11 ianuarie 2010 17:11:10
Problema Algoritmul lui Dijkstra Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 1.24 kb
using namespace std;
//#include<iostream>
#include<fstream>
#define NN 50005
#define INFINIT 1<<30
int n,v[NN],d[NN];//t[NN];
struct nod
{
	int capat,cost;
	nod* next;
};
nod* G[NN];
void init(int sursa)
{
	for(int i=0;i<=n;++i)
		v[i]=0,d[i]=INFINIT;//t[i]=1;
	v[sursa]=1,d[sursa]=0;//t[sursa]=0;
	for(nod *p=G[sursa]; p; p=p->next)
		d[p->capat]=p->cost;
		//,t[p->capat]=sursa;
}
void dijkstra(int sursa)
{
	init(sursa);
	int gata=0;
	while(!gata)
	{
		int pmin=0;
		for(int i=1;i<=n;++i)
			if(v[i]==0 && d[i]<d[pmin])
				pmin=i;
		if(pmin==0)
			gata=1;
		else
		{
			v[pmin]=1;
			for(nod* p=G[pmin];p;p=p->next)
				if(v[p->capat]==0)
					if(d[p->capat]>d[pmin]+p->cost)
						d[p->capat]=d[pmin]+p->cost;// t[p->capat]=pmin;
		}
	}
}
void addarc(int i,int j,int c)
{
	nod *p=new nod;
	p->capat=j;
	p->cost=c;
	p->next=G[i];
	G[i]=p;
}
int main()
{
	int i,j,cost,m;
	ifstream fin("dijkstra.in");
	fin>>n>>m;
	for(;m;--m)
	{
		fin>>i>>j>>cost;
		addarc(i,j,cost);
	}
	fin.close();
	dijkstra(1);
	/*for(int i=1;i<=n;++i)
		cout<<t[i]<<' ';*/
	ofstream fout("dijkstra.out");
	for(int i=2;i<=n;++i)
		if(d[i]==INFINIT)
			fout<<0<<' ';
		else
		   fout<<d[i]<<' ';
	fout.close();
	return 0;
}