Cod sursa(job #590693)

Utilizator lily3Moldovan Liliana lily3 Data 19 mai 2011 14:03:07
Problema Distante Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.3 kb
#include<fstream>
#define inf 1000000
using namespace std;

int i,j,n,m,t,s,c1[50001],x,y,c;
int *l1[50001],*l2[50001],d[50001],q[1000001];
void rez()
{
	int i,st,dr,p;
	d[s]=0;
	q[0]=s;
	st=dr=0;
	while(st<=dr)
	{
		p=q[st++];
		for(i=1;i<=l1[p][0];i++)
			if(d[l1[p][i]]>d[p]+l2[p][i])
				d[l1[p][i]]=d[p]+l2[p][i],q[++dr]=l1[p][i];
	}
}
int main()
{
	FILE *f=fopen("distante.in","r");
	FILE *g=fopen("distante.out","w");
	fscanf(f,"%d",&t);
	while(t--)
	{
		fscanf(f,"%d%d%d",&n,&m,&s);
		for(i=1;i<=n;i++)
		{
			fscanf(f,"%d",&c1[i]);
			d[i]=inf;
			l1[i]=(int *)realloc(l1[i],sizeof(int));
			l1[i][0]=0;
			l2[i]=(int *)realloc(l2[i],sizeof(int));
			l2[i][0]=0;
		}
		for(i=1;i<=m;i++)
		{
			fscanf(f,"%d%d%d",&x,&y,&c);
			l1[x][0]++;
			l1[x]=(int *)realloc(l1[x],(l1[x][0]+1)*sizeof(int));
			l1[x][l1[x][0]]=y;
			l1[y][0]++;
			l1[y]=(int *)realloc(l1[y],(l1[y][0]+1)*sizeof(int));
			l1[y][l1[y][0]]=x;
			l2[x][0]++;
			l2[x]=(int *)realloc(l2[x],(l2[x][0]+1)*sizeof(int));
			l2[x][l2[x][0]]=c;
			l2[y][0]++;
			l2[y]=(int *)realloc(l2[y],(l2[y][0]+1)*sizeof(int));
			l2[y][l2[y][0]]=c;
		}
		rez();
		int ok=1;
		for(i=1;i<=n;i++)
			if(c1[i]!=d[i])
				ok=0,i=n+1;
			if(ok)
				fprintf(g,"DA\n");
			else
				fprintf(g,"NU\n");
	}
	return 0;
}