Pagini recente » Cod sursa (job #420682) | Cod sursa (job #2521621) | Cod sursa (job #2699012) | Cod sursa (job #3181939) | Cod sursa (job #2927827)
#include <iostream>
#include <fstream>
#include <algorithm>
#include <queue>
#include <set>
#define pr pair<int, int>
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
const int N = 5e4 + 7, inf = 1e9;
int d[N];
vector <pr> v[N];
set <pr> s;
void dijkstra(int x, int n){
for(int i = 1; i <= n; i++)
d[i] = inf;
d[x] = 0; s.insert({0, x});
while(!s.empty()){
pr y;
y = *s.begin();
s.erase(s.begin());
int nod = y.second;
for(pr i : v[nod]){
int newnod = i.second, cost = i.first;
if(d[newnod] > d[nod] + cost){
d[newnod] = d[nod] + cost;
s.insert(i);
}
}
}
}
int main()
{
int T;
fin >> T;
while(T--){
int n, m, p;
fin >> n >> m >> p;
int dr[N];
for(int i = 1; i <= n; i++)
fin >> dr[i];
while(m--){
int i, j, c;
fin >> i >> j >> c;
v[i].push_back({c, j});
v[j].push_back({c, i});
}
dijkstra(p, n);
bool ok = true;
for(int i = 1; i <= n; i++){
if(d[i] != dr[i]){
ok = false;
break;
}
}
if(ok) fout << "DA\n";
else fout << "NU\n";
}
return 0;
}