Pagini recente » Cod sursa (job #1173115) | Cod sursa (job #454377) | Cod sursa (job #636341) | Cod sursa (job #956730) | Cod sursa (job #869339)
Cod sursa(job #869339)
#include <cstdio>
#include <vector>
#include <queue>
#define N 50001
#define infinit 999999999
using namespace std;
struct nod
{
int v;
int c;
};
int n,m,t,s;
int d[N],rezultat[N];
struct cmp
{
bool operator()(const int &a, const int &b)const
{
return d[a]>d[b];
}
};
vector<nod>h[N];
priority_queue<int,vector<int>,cmp>q;
void citire()
{
scanf("%d %d %d",&n,&m,&s);
for(int i=1;i<=n;i++)
scanf("%d",&rezultat[i]);
for(int i=1;i<=m;i++)
{
int v;
nod a;
scanf("%d %d %d",&v,&a.v,&a.c);
h[v].push_back(a);
}
}
void dijkstra()
{
d[s]=0;
for(int i=1;i<=n;i++)
{
if(i!=s)
d[i]=infinit;
}
q.push(s);
while(!q.empty())
{
int v=q.top();
q.pop();
int nr=h[v].size();
for(int i=0;i<nr;i++)
{
nod p=h[v][i];
if(d[v]+p.c<d[p.v])
{
d[p.v]=d[v]+p.c;
q.push(p.v);
}
}
}
}
void afisare()
{
for(int i=1;i<=n;i++)
{
if(d[i]!=rezultat[i])
{
printf("NU\n");
return;
}
}
printf("DA\n");
}
int main()
{
freopen("distante.in","r",stdin);
freopen("distante.out","w",stdout);
scanf("%d",&t);
for(int i=0;i<t;i++)
{
citire();
dijkstra();
afisare();
}
return 0;
}