Cod sursa(job #258171)

Utilizator ada_sAda-Mihaela Solcan ada_s Data 14 februarie 2009 20:04:06
Problema Sate Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.08 kb
#include <stdio.h>
#include <string.h>

const int NMAX=2*30010;
const int MMAX=2*100050;

long *c[NMAX], tc[MMAX], m, dist[NMAX];
int x[MMAX], y[MMAX], n, ax, ay, cont[NMAX], q[NMAX], *a[NMAX];

long bf(int x, int y);

int main()
{
	long i;
	freopen("sate.in", "r", stdin);
	freopen("sate.out", "w", stdout);
	scanf("%d%ld%d%d", &n, &m, &ax, &ay);
	for (i=0; i<m; i++)
	{
		scanf("%d%d%ld", &x[i], &y[i], &tc[i]);
		cont[x[i]]++;
		cont[y[i]]++;
	}//for i
	for (i=1; i<=n; i++)
	{
		a[i]=new int[cont[i]+1];
		c[i]=new long[cont[i]+1];
		a[i][0]=0;
		c[i][0]=0;
	}//for i
	for (i=0; i<m; i++)
	{
		a[x[i]][++a[x[i]][0]]=y[i];
		a[y[i]][++a[y[i]][0]]=x[i];
		c[x[i]][++c[x[i]][0]]=tc[i];
		c[y[i]][++c[y[i]][0]]=-tc[i];
	}//for i
	printf("%ld", bf(ax, ay));
	return 0;
}//main

long bf(int x, int y)
{
	int u=0, p=0, i, t;
	memset(dist, -1, sizeof(dist));
	q[u]=x;
	dist[x]=0;
	while (p<=u)
	{
		t=q[p++];
		for (i=1; i<=a[t][0]; i++)
			if (dist[a[t][i]]<0)
			{
				dist[a[t][i]]=dist[t]+c[t][i];
				q[++u]=a[t][i];
				if (a[t][i]==y)
					return dist[y];
			}//if				
	}//while
	return -1;
}//bf