Pagini recente » Cod sursa (job #660277) | Cod sursa (job #1363256) | Cod sursa (job #2261094) | Cod sursa (job #2266792) | Cod sursa (job #1977134)
#include <iostream>
#include <fstream>
#include <set>
#include <vector>
#define N 50010
#define INF 20000010
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
int i,j,n,s,m,k,t,p,r,dist[N],sol[N];
vector < pair<int,int> > vec[N];
set < pair<int,int> > h;
int main()
{
int x,y,z,nod,to,cost,ok;
f>>t;
while(t!=0)
{
f>>n>>m>>s;
for(int i=1;i<=n;i++)
f>>sol[i];
for(int i=1;i<=n;i++){
dist[i]=INF;
vec[i].clear();
}
while(m!=0)
{
f>>x>>y>>z;
vec[x].push_back(make_pair(y,z));
vec[y].push_back(make_pair(x,z));
m--;
}
ok=1;
dist[s]=0;
h.insert(make_pair(0,s));
while(!h.empty())
{
nod=h.begin()->second;
h.erase(h.begin());
for(vector< pair<int,int> >::iterator it=vec[nod].begin();it!=vec[nod].end();it++)
{
to=it->first;
cost=it->second;
if(dist[to]>dist[nod]+cost){
if(dist[to]!=INF)
{
h.erase(h.find(make_pair(dist[to],to)));
}
dist[to]=dist[nod]+cost;
h.insert(make_pair(dist[to],to));
}
}
}
for(int i=1;i<=n;i++)
{
if(dist[i]==INF)
dist[i]=0;
if(dist[i]!=sol[i])
{
ok=0;
}
}
if(ok==1)
g<<"DA";
else
g<<"NU";
g<<'\n';
t--;
}
}