Pagini recente » Cod sursa (job #529146) | Monitorul de evaluare | Cod sursa (job #1538133) | Cod sursa (job #2858408) | Cod sursa (job #2947520)
#include <iostream>
#include <fstream>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
ifstream fin ("distante.in");
ofstream fout ("distante.out");
const int NMAX=5e4+5;
const int INF=1e9;
int dist[NMAX];
int viz[NMAX];
int aux[NMAX];
priority_queue<pair<int,int>>q;
vector<pair<int,int>>v[NMAX];
void dijkstra(int p)
{
dist[p]=0;
viz[p]=0;
q.push(make_pair(0,p));
while(!q.empty())
{
p=q.top().second;
q.pop();
viz[p]=0;
for(auto i:v[p])
{
if(dist[p]+i.second<dist[i.first])
{
dist[i.first]=dist[p]+i.second;
if(viz[i.first]==0)
{
viz[i.first]=1;
q.push(make_pair(-dist[i.first],i.first));
}
}
}
}
}
int main()
{
int n,m,s,t;
fin>>t;
while(t--)
{
bool ok=true;
fin>>n>>m>>s;
for(int i=1;i<=n;i++)
{
v[i].clear();
viz[i]=0;
dist[i]=INF;
fin>>aux[i];
}
while(m--)
{
int a,b,c;
fin>>a>>b>>c;
v[a].push_back(make_pair(b,c));
v[b].push_back(make_pair(a,c));
}
dijkstra(s);
for(int i=1;i<=n;i++)
{
if(dist[i]!=aux[i])
{
ok=false;
break;
}
}
if(ok)
fout<<"DA\n";
else
fout<<"NU\n";
}
return 0;
}