Pagini recente » Cod sursa (job #2943037) | Cod sursa (job #1914495) | Cod sursa (job #2347140) | Cod sursa (job #2649296) | Cod sursa (job #2858121)
#include <bits/stdc++.h>
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
int n,m,i,dist[500005],a,b,val,nod,vecin,viz[500005],sol[500005],t,ok,s;
priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>>Q;
vector <pair<int,int>> v[50005];
int main()
{
//lungime 0 nu infinit
fin>>t;
while(t--){
ok = 0;
fin>>n>>m>>s;
for(i=1;i<=n;i++){
fin>>sol[i];
}
for(i=1;i<=m;i++){
fin>>a>>b>>val;
v[a].push_back({b,val});
v[b].push_back({a,val});
}
dist[s] = 0;
//invers coada
Q.push({0,s});
for(i=1;i<=n;i++){
if(i != s)
dist[i] = 200000000;
else
dist[i] = 0;
viz[i] = 0;
}
while(!Q.empty()){
nod = Q.top().second;
val = Q.top().first;
Q.pop();
if(!viz[nod]){
for(i=0;i<v[nod].size();i++){
vecin = v[nod][i].first;
if(dist[vecin] > val + v[nod][i].second){
dist[vecin] = val + v[nod][i].second;
Q.push({dist[vecin],vecin});
}
}
viz[nod] = 1;
}
}
for(i=2;i<=n;i++){
if(dist[i] == 200000000)
dist[i] = 0;
if(dist[i] != sol[i]){
ok = 1;
fout << "NU" << '\n';
break;
}
}
if(!ok)
fout << "DA" << '\n';
}
return 0;
}