Cod sursa(job #674133)

Utilizator okros_alexandruOkros Alexandru okros_alexandru Data 5 februarie 2012 17:12:16
Problema Sate Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <fstream>
#include <vector>
#define NMAx 30100
#define pb push_back
using namespace std;
vector <int> G[NMAx],D[NMAx];
int n,nr,X,Y,dist[NMAx],deque[NMAx];

void citire() {
	
	int i,m,x,y,c;
	ifstream in("sate.in");
	in>>n>>m>>X>>Y;
	for(i=0;i<m;i++)
		{in>>x>>y>>c;
		if(x>y) c=-c;
		G[x].pb(y);
		G[y].pb(x);
		D[x].pb(c);
		D[y].pb(-c);
		}
	in.close();
	
}
void BFS() {
	
	int i,j,nod,vec;
	
	if(X>Y)
		swap(X,Y);
	deque[1]=X;
	dist[X]=1;
	nr=1;
	
	for(i=1;i<=nr;i++) {
		nod=deque[i];
		for(j=0;j<G[nod].size();j++) {
			vec=G[nod][j];
			if(!dist[vec]) {
				deque[++nr]=vec;
				dist[vec]=dist[nod]+D[nod][j];
				}
			}
		}
	
}
int main()
{
	
	citire();
	BFS();
	
	ofstream out("sate.out");
	out<<(dist[Y]-1)<<'\n';
	out.close();
	
	return 0;

}