Pagini recente » Cod sursa (job #2643726) | Cod sursa (job #1638695) | Cod sursa (job #724384) | Cod sursa (job #2990502) | Cod sursa (job #590693)
Cod sursa(job #590693)
#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;
}