Pagini recente » Cod sursa (job #503350) | Cod sursa (job #57645) | Cod sursa (job #17650) | Cod sursa (job #1518921) | Cod sursa (job #2583032)
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
int teste, start;
int n, m, i, j, x, y, c, d[50010], dPresupus[50010], v[50010], nod, vecin, cost;
vector < pair<int, int> > a[50010];
queue <int> s;
bool bellman(){
for(i=1; i<=n; i++){
d[i]=1000000000;
v[i]=0;
}
s.push(start);
v[start]=1;
d[start]=0;
while(!s.empty()){
nod=s.front();
v[nod]=0;
for(i=0; i<a[nod].size(); i++){
vecin=a[nod][i].first;
cost=a[nod][i].second;
if(d[nod]+cost<d[vecin]){
d[vecin]=d[nod]+cost;
if(v[vecin]==0){
s.push(vecin);
v[vecin]=1;
}
}
}
s.pop();
}
for(i=1; i<=n; i++){
if(d[i]!=dPresupus[i]){
return 0;
}
}
return 1;
}
int main(){
fin>>teste;
for(; teste; teste--){
fin>>n>>m>>start;
for(i=1; i<=n; i++){
fin>>dPresupus[i];
a[i].clear();
}
for(i=1; i<=m; i++){
fin>>x>>y>>c;
a[x].push_back({y, c});
a[y].push_back({x, c});
}
if(bellman()==1){
fout<<"DA"<<"\n";
}else{
fout<<"NU"<<"\n";
}
}
}