Pagini recente » Cod sursa (job #2174368) | Cod sursa (job #385314) | Cod sursa (job #75881) | Cod sursa (job #2157519) | Cod sursa (job #2276864)
#include<bits/stdc++.h>
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
int t,n,m,s,dr[50001],dn[50001];
vector< pair<int,int> >G[50001];
priority_queue< pair< int,int >, vector< pair<int,int> >,greater< pair<int,int> > >Q;
bool viz[50001];
void dij(int start)
{
for(int i=1;i<=n;++i)
dn[i]=INT_MAX;
dn[start]=0;
Q.push({0,start});
while(!Q.empty())
{
int nodCurent=Q.top().second;
Q.pop();
if(viz[nodCurent])
continue;
viz[nodCurent]=1;
for(int i=0;i<G[nodCurent].size();++i)
{
int cost=G[nodCurent][i].second;
int nodveci=G[nodCurent][i].first;
if(cost+dn[nodCurent]<dn[nodveci])
{
dn[nodveci]=cost+dn[nodCurent];
}
if(!viz[nodveci])
{
Q.push({dn[nodveci],nodveci});
}
}
}
}
int main()
{
f>>t;
while(t--)
{
f>>n>>m>>s;
memset(viz,0,sizeof(viz));
for(int i=1;i<=n;++i)
{
f>>dr[i];
G[i].clear();
}
for(int i=1;i<=m;++i)
{
int x,y,c;
f>>x>>y>>c;
G[x].push_back({y,c});
G[y].push_back({x,c});
}
dij(s);
bool ok=1;
for(int i=1;i<=n && ok;++i)
{
if(dr[i]!=dn[i])
{
ok=0;
}
}
if(ok==0)
g<<"NU\n";
else
g<<"DA\n";
}
}