Pagini recente » Cod sursa (job #630717) | Cod sursa (job #278041) | Cod sursa (job #2910232) | Cod sursa (job #1571393) | Cod sursa (job #892420)
Cod sursa(job #892420)
#include<fstream>
#define inf 1000000000
#define NMax 50003
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
struct Nod
{
int nod, cost;
Nod *next;
};
Nod *t,*Vecin[NMax];
int j,m,n,x,y,costul,nod,d[NMax],coada[5*NMax];
void adauga(int x, int y, int costul)
{
Nod *q = new Nod;
q->nod = y;
q->cost = costul;
q->next = Vecin[x];
Vecin[x] = q;
}
void bellman_ford(int k){
int p,u,i;
for(i=1;i<=n;i++)
d[i]=inf;
p=1;
u=1;
coada[u]=k;
d[k]=0;
while (p<=u)
{
j=coada[p];p++;
t=Vecin[j];
while(t)
{
if(d[t->nod]>d[j]+t->cost)
{
coada[++u]=t->nod;
d[t->nod]=d[j]+t->cost;
}
t=t->next;
}
}
}
int main()
{
int i,k,zaharel[10000],T,ok;
f>>T;
for(int j=1; j<=T; j++)
{
f>>n>>m>>k;
for(i=1; i<=n; i++)
f>>zaharel[i];
for(i=1;i<=m;i++)
{
f>>x>>y>>costul;
adauga(x,y,costul);
}
bellman_ford(k);
ok=0;
for ( int i = 1; i <= n; i++ )
if((d[i] == inf && zaharel[i]==0)||(d[i]==zaharel[i]))
ok=0;
else
ok=1;
if(ok==0)
g<<"DA\n";
else
g<<"NU\n";
}
f.close();
g.close();
return 0;
}