Pagini recente » Cod sursa (job #594125) | Cod sursa (job #1670360) | Cod sursa (job #604347) | Cod sursa (job #1440676) | Cod sursa (job #1316500)
#include<iostream>
#include<fstream>
#include<queue>
#include<vector>
#define inf 0x3f3f3f3f
using namespace std;
priority_queue<pair<int,int> , vector<pair<int, int> > , greater<pair<int,int> > > q;
vector<pair<int, int > > v[50004];
int i,n,d[50004],m,sol[50004],t,shp,s,ok,a,b,c,u,x;
int main()
{
ifstream cin("distante.in");
ofstream cout("distante.out");
cin>>t;
for(shp=1;shp<=t;shp++)
{
cin>>n>>m>>s;
for(i=1;i<=n;i++)
{
cin>>sol[i];
d[i]=inf;
v[i].clear();
}
for(i=1;i<=m;i++)
{
cin>>a>>b>>c;
v[a].push_back(make_pair(b,c));
v[b].push_back(make_pair(a,c));
}
d[s]=0;
q.push(make_pair(0,s));
while(!q.empty())
{
c=q.top().first;
x=q.top().second;
q.pop();
for(vector<pair<int,int> > :: iterator it=v[x].begin();it!=v[x].end();it++)
{
if(d[it->first]>c+it->second)
{
d[it->first]=c+it->second;
q.push(make_pair(d[it->first],it->first));
}
}
}
ok=0;
for(i=1;i<=n;i++)
{
if(d[i]==sol[i]||(d[i]==inf && sol[i]==0))
continue;
ok=1;
}
if(ok==0)
cout<<"DA\n";
else cout<<"NU\n";
}
return 0;
}