Pagini recente » Cod sursa (job #2584254) | Cod sursa (job #1860551) | Cod sursa (job #1032082) | Cod sursa (job #1931623) | Cod sursa (job #2302926)
#include <bits/stdc++.h>
#define NMAX 50005
#define inf 1e9
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
vector<pair<int,int> >v[NMAX];
vector<int>sol;
priority_queue<pair<int,int>,vector<pair<int,int> >,greater<pair<int,int> > >h;
bool use[NMAX];
int d[NMAX],i,j,n,m,p,s,sl,nod,vecin,ok,x,y,c;
int main()
{
f>>p;
for(i=1;i<=p;i++)
{
f>>n>>m>>s;
for(j=1;j<=n;j++)
{
f>>sl;
sol.push_back(sl);
}
for(j=1;j<=m;j++)
{
f>>x>>y>>c;
v[x].push_back({y,c});
v[y].push_back({x,c});
}
for(j=1;j<=n;j++)
{
d[j]=inf;
}
d[s]=0;
h.push({0,s});
while(!h.empty())
{
nod=h.top().second;
h.pop();
for(j=0;j<v[nod].size();j++)
{
vecin=v[nod][j].first;
if(d[nod]+v[nod][j].second<d[vecin])
{
d[vecin]=d[nod]+v[nod][j].second;
h.push({d[vecin],vecin});
}
}
}
ok=1;
for(j=0;j<sol.size();j++)
{
if(sol[j]!=d[j+1]){ok=0;break;}
}
if(ok==1)g<<"DA"<<'\n';
else g<<"NU"<<'\n';
sol.clear();
for(j=1;j<=n;j++)
v[j].clear();
}
return 0;
}