Pagini recente » Cod sursa (job #369969) | Cod sursa (job #2862783) | Cod sursa (job #2034117) | Monitorul de evaluare | Cod sursa (job #2654278)
#include <bits/stdc++.h>
#define inf 2000000000
#define DIM 50010
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
int t,n,m,i,d[DIM],D[DIM],s,nod,vecin,cost,x,y,c;
set< pair<int,int> > S;
vector< pair<int,int> > L[DIM];
int dijkstra(int s) {
for (int i=1;i<=n;i++)
d[i]=inf;
d[s]=0;
S.insert({0,s});
while (!S.empty()) {
nod=S.begin()->second;
S.erase(S.begin());
for (auto it:L[nod]) {
vecin=it.first, cost=it.second;
if (d[vecin]>d[nod]+cost) {
S.erase({d[vecin],vecin});
d[vecin]=d[nod]+cost;
if (d[vecin]<D[vecin])
return 0;
S.insert({d[vecin],vecin});
}
}
}
return 1;
}
int main() {
fin>>t;
while (t--) {
fin>>n>>m>>s;
for (i=1;i<=n;i++)
fin>>D[i];
/*if (D[s]!=0) {
fout<<"NU\n";
continue;
}*/
for (i=1;i<=n;i++)
L[i].clear();
S.clear();
for (i=1;i<=m;i++) {
fin>>x>>y>>c;
L[x].push_back({y,c});
L[y].push_back({x,c});
}
if (!dijkstra(s)) {
fout<<"NU\n";
continue;
}
int ok=1;
for (i=1;i<=n;i++)
if (d[i]!=D[i]) {
fout<<"NU\n";
ok=0;
break;
}
if (ok)
fout<<"DA\n";
}
return 0;
}