Cod sursa(job #651570)

Utilizator BarracudaFMI-Alex Dobrin Barracuda Data 20 decembrie 2011 20:56:47
Problema Sate Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.79 kb
#include<cstdio>
#include<vector>
#define NMAX 100025
using namespace std;
int vaf,p,u,n,m,x,y,q,w,d,viz[NMAX],C[NMAX],dist[NMAX],i,nod;
vector<int >G[NMAX],K[NMAX];
int main (){
	freopen("sate.in","r",stdin);
	freopen("sate.out","w",stdout);
	scanf("%d%d%d%d",&n,&m,&x,&y);
	for(i=1;i<=m;i++){
		scanf("%d%d%d",&q,&w,&d);
		G[q].push_back(w);
		G[w].push_back(q);
		if(q>w){
			K[q].push_back(-d);
			K[w].push_back(d);
		}
		else{
			K[q].push_back(d);
			K[w].push_back(-d);
		}
	}
	C[1]=x;
	viz[x]=1;
	for(p=u=1;p<=u;p++){
		nod=C[p];
		for(size_t i=0;i<G[nod].size();++i){
			if(!viz[G[nod][i]]){
				C[++u]=G[nod][i];
				dist[G[nod][i]]+=dist[nod]+K[nod][i];
				viz[G[nod][i]]=1;
			}
			if(G[nod][i]==y)
				p=u+1;
		}
	}
	printf("%d",dist[y]);
	return 0;
}