Cod sursa(job #1340186)

Utilizator afkidStancioiu Nicu Razvan afkid Data 11 februarie 2015 17:30:20
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.23 kb
#include <cstdio>
#include <vector>
#include <queue>
#include <utility>
#define N 50010
#define INF 0x3f3f3f3f


using namespace std;

int n,m;
vector<pair<int,int> > arcs[N];
int d[N];
int pred[N];

int readInt()
{
	int n=0;
	char c=getchar();
	while(c<'0' || c>'9') c=getchar();
	while(c>='0' && c<='9')
	{
		n=(n<<1)+(n<<3) +c-'0';
		c=getchar();
	}
	return n;
}

void Read()
{
	freopen("dijkstra.in","r",stdin);
	int a,b,d;
	n=readInt();
	m=readInt();
	for(int i=0;i<m;++i)
	{
		a=readInt();
		b=readInt();
		d=readInt();
		arcs[a].push_back(make_pair(b,d));
	}
}

void Dijkstra(int s)
{

}

void BellmanFord(int s)
{
	queue<int> fifo;
	for(int i=1;i<=n;++i)
	{
		d[i]=INF;
	}
	d[s]=0;
	fifo.push(s);
	vector<pair<int,int> >::iterator it;
	while(!fifo.empty())
	{
		int p=fifo.front();
		for(it=arcs[p].begin();it!=arcs[p].end();++it)
		{
			if(d[it->first]>d[p]+it->second)
			{
				d[it->first]=d[p]+it->second;
				fifo.push(it->first);	
			}
		}
		fifo.pop();
	}
}

void Write()
{
	freopen("dijkstra.out","w",stdout);
	for(int i=2;i<=n;++i)
	{
		printf("%d ", (d[i]<INF)? d[i] : 0);
	}
	printf("\n");
}

int main()
{
	Read();
	BellmanFord(1);
	Write();
	return 0;
}