Pagini recente » Cod sursa (job #600327) | Cod sursa (job #3135237) | Cod sursa (job #558810) | Cod sursa (job #2608986) | Cod sursa (job #2476762)
#include <bits/stdc++.h>
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
typedef pair<int,int> graf;
vector <graf> G[50005];
priority_queue <graf,vector<graf>,greater<graf>> h;
int t,n,m,x,y,i,j,d[50005],dc[50005],pl,c;
bool sel[50005];
void dijkstra(int pl)
{
for(int i=1;i<=n;i++)
{
d[i]=INT_MAX;
sel[i]=false;
}
sel[pl]=true;
d[pl]=0;
for(auto it : G[pl])
{
h.push(it);
d[it.second]=it.first;
}
while(!h.empty())
{
while(!h.empty() && sel[h.top().second])
h.pop();
if(h.empty())
return;
graf k=h.top();
sel[k.second]=true;
h.pop();
for(auto it : G[k.second])
{
if(k.first+it.first<d[it.second])
{
d[it.second]=k.first+it.first;
h.push({d[it.second],it.second});
}
}
}
}
int main()
{
f>>t;
for(j=1;j<=t;j++)
{
f>>n>>m>>pl;
for(i=1;i<=n;i++)
f>>dc[i];
for(i=1;i<=n;i++)
G[i].clear();
for(i=1;i<=m;i++)
{
f>>x>>y>>c;
G[x].push_back({c,y});
G[y].push_back({c,x});
}
dijkstra(pl);
bool ok=true;
for(i=1;i<=n;i++)
if(d[i]!=dc[i])
ok=false;
if(ok==true)
g<<"DA"<<'\n';
else g<<"NU"<<'\n';
}
return 0;
}