Pagini recente » Cod sursa (job #279757) | Cod sursa (job #1488742) | Cod sursa (job #2431723) | Cod sursa (job #1307305) | Cod sursa (job #2655659)
#include <bits/stdc++.h>
using namespace std;
ifstream fin ("distante.in");
ofstream fout ("distante.out");
int n, m, i, a, b, c, S, teste, t, nod, ok, cost, vecin;
int d[50005], check[50005];
vector <pair<int, int> > L[50005];
set <pair <int, int> > s;
int main(){
fin >> t;
for (teste=1; teste<=t; teste++){
fin >> n >> m >> S;
//////
for (i=1; i<=n; i++){
L[i].clear();
}
s.clear();
for (i=1; i<=n; i++){
d[i] = INT_MAX;
}
d[S] = 0;
ok = 0;
//////
for (i=1; i<=n; i++){
fin >> check[i];
}
for (i=1; i<=m; i++){
fin >> a >> b >> c;
L[a].push_back ({b, c});
L[b].push_back ({a, c});
}
s.insert ({0, S});
while (!s.empty()){
nod = s.begin()->second;
if (d[nod] != check[nod]){
ok = 1;
break;
}
s.erase(s.begin());
for (i=0; i<L[nod].size(); i++){
vecin = L[nod][i].first, cost = L[nod][i].second;
if (d[nod] + cost < d[vecin]){
s.erase ({d[vecin], vecin});
d[vecin] = d[nod] + cost;
s.insert ({d[vecin], vecin});
}
}
}
for (i=1; i<=n; i++){
if (d[i] != check[i]){
ok = 1;
break;
}
}
if (ok == 0)
fout << "DA\n";
else
fout << "NU\n";
}
return 0;
}