Pagini recente » Cod sursa (job #1880063) | Cod sursa (job #1575700) | Cod sursa (job #2087091) | Cod sursa (job #724547) | Cod sursa (job #271019)
Cod sursa(job #271019)
#include<fstream.h>
#define N 101
#define INF 0x3f3f3f3f
ifstream fin("distante.in");
ofstream fout("distante.out");
struct nod{
int info,cost;
nod *next;
};
nod *prim[N];
int t,n,m,s,d[N],sol[N],qu[N],viz[N];
void insert(int where, int what, int how_much)
{ nod *p;
p=new nod;
p->info=what;
p->cost=how_much;
p->next=prim[where];
prim[where]=p;
}
void bellmanFord()
{ nod *q;
int p,u,X;
memset(d,INF,N);
memset(viz,0,N);
p=u=1;
viz[s]=1;
d[s]=0;
qu[p]=s;
while(p<=u)
{ X=qu[p];viz[X]=0;
for(q=prim[X];q;q=q->next)
if(d[q->info]>d[X]+q->cost)
{ d[q->info]=d[X]+q->cost;
if(!viz[q->info])
{ qu[++u]=q->info;
viz[q->info]=1;
}
}
p++;
}
}
int check()
{ int i;
for(i=1;i<=n;i++)
if(d[i]!=sol[i]) return 0;
return 1;
}
int main()
{ int i,x,y,ct;
fin>>t;
while(t)
{ fin>>n>>m>>s;
for(i=1;i<=n;i++)
fin>>sol[i];
for(i=1;i<=m;i++)
{ fin>>x>>y>>ct;
insert(x,y,ct);
insert(y,x,ct);
}
bellmanFord();
if(check())
fout<<"DA\n";
else fout<<"NU\n";
t--;
}
return 0;
}