Pagini recente » Cod sursa (job #381063) | Cod sursa (job #1525025) | Cod sursa (job #1666866) | Cod sursa (job #2476983) | Cod sursa (job #2484869)
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>
#define inf 0x3f3f3f3f
#define pb push_back
#define nmax 50001
using namespace std;
ifstream in("distante.in");
ofstream out("distante.out");
struct eee
{
int nod,cost;
bool operator <(const eee &a)const
{
return cost>a.cost;
}
};
int n,m,s,d[nmax],dmin[nmax],a,b,c,t;
vector <eee> v[nmax];
priority_queue<eee> q;
void dijkstra(int nd)
{
int x,y;
memset(dmin,inf,sizeof dmin);
dmin[nd]=0;
q.push({nd,0});
while(!q.empty())
{
x=q.top().nod;
y=q.top().cost;
q.pop();
if(y!=dmin[x])continue;
for(auto i:v[x])
{
if(dmin[i.nod]>dmin[x]+i.cost)
{
dmin[i.nod]=dmin[x]+i.cost;
q.push({i.nod,dmin[i.nod]});
}
}
}
}
int main()
{
in>>t;
for(int i=1;i<=t;i++)
{
in>>n>>m>>s;
for(int j=1;j<=n;j++)
{
in>>d[j];
}
for(int j=1;j<=m;j++)
{
in>>a>>b>>c;
v[a].pb({b,c});
v[b].pb({a,c});
}
dijkstra(s);
int ok=1;
for(int j=1;j<=n;j++)
{
if(d[j]!=dmin[j])ok=0;
v[j].clear();
}
if(ok)
out<<"DA";
else
out<<"NU";
out<<'\n';
}
return 0;
}