Pagini recente » Cod sursa (job #2840613) | Cod sursa (job #2638313) | Cod sursa (job #2176335) | Cod sursa (job #725291) | Cod sursa (job #2825268)
#include <bits/stdc++.h>
#include <iostream>
#include <fstream>
#include <vector>
#include <climits>
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
int n,m,start,T;
bool vizitat[50001];
int distanta[50001];
vector <pair <int, int> > vecini[50001];
void dijkstra(int start){
priority_queue <pair<int, int>, vector <pair<int,int> >, greater <pair<int,int> > > Q;
Q.push(make_pair(0,start));
while(Q.empty()==0){
int nod_curent=Q.top().second;
Q.pop();
if(vizitat[nod_curent]==0){
vizitat[nod_curent]=1;
for(int i=0;i<int(vecini[nod_curent].size());i++){
int vecin=vecini[nod_curent][i].first;
if(distanta[vecin]>distanta[nod_curent]+vecini[nod_curent][i].second){
distanta[vecin]=distanta[nod_curent]+vecini[nod_curent][i].second;
Q.push(make_pair(distanta[vecin], vecin));
}
}
}
}
}
int main()
{
f>>T;
for(int i=1;i<=T;i++){
f>>n>>m>>start;
int distante_bronzanel[n+1];
for(int j=1;j<=n;j++){
f>>distante_bronzanel[j];
}
for(int j=1;j<=n;j++){
vizitat[j]=0;
}
int val_max=INT_MAX;
for(int j=1;j<=n;j++){
distanta[j]=val_max;
}
distanta[start]=0;
for(int j=1;j<=m;j++){
int a,b,c;
f>>a>>b>>c;
vecini[a].push_back(make_pair(b,c));
vecini[b].push_back(make_pair(a,c));
}
dijkstra(start);
for(int j=1;j<=n;j++){
vecini[j].erase(vecini[j].begin(),vecini[j].end());
}
bool OK=1;
for(int j=1;j<=n;j++){
if(distanta[j]!=distante_bronzanel[j]){
OK=0;
break;
}
}
if(OK==1){
g<<"DA\n";
}
else
g<<"NU\n";
}
return 0;
}