Pagini recente » Cod sursa (job #2851913) | Cod sursa (job #143593) | Cod sursa (job #3163139) | Cod sursa (job #827029) | Cod sursa (job #2569143)
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
#define NMax 50005
#define oo (1 << 30)
int N,M,S,numar;
int D[NMax],B[NMax];
bool incoada[NMax];
struct compara
{
bool operator()(int x,int y)
{
return D[x]>D[y];
}
};
vector< pair <int,int> >V[NMax];
priority_queue < int, vector<int>,compara> Q;
void read()
{
f>>N>>M>>S;
for(int i=1; i<=N; i++)
f>>B[i];
for(int i=1; i<=M; i++)
{
int x,y,c;
f>>x>>y>>c;
V[x].push_back(make_pair(y,c));
}
}
void write()
{
bool ok=1;
for(int i=1; i<=N; i++)
if(D[i]!=B[i])
ok=0;
if(ok==0)
g<<"NU"<<'\n';
else
g<<"DA"<<'\n';
}
void Dijkstra(int nodStart)
{
for(int i=1; i<=N; i++)
D[i]=oo;
D[nodStart]=0;
Q.push(nodStart);
incoada[nodStart]=true;
while(!Q.empty())
{
int nodCurent=Q.top();
Q.pop();
incoada[nodCurent]=false;
for(size_t i=0; i<V[nodCurent].size(); i++)
{
int Vecin = V[nodCurent][i].first;
int Cost = V[nodCurent][i].second;
if(D[nodCurent]+Cost<D[Vecin])
{
D[Vecin]=D[nodCurent]+Cost;
if(incoada[Vecin]==false)
{
incoada[Vecin]=true;
Q.push(Vecin);
}
}
}
}
}
int main()
{
f>>numar;
for(int miaw=1;miaw<=numar;miaw++)
{
read();
Dijkstra(S);
write();
}
return 0;
}