Pagini recente » Cod sursa (job #1827337) | Cod sursa (job #1795008) | Cod sursa (job #300159) | Cod sursa (job #1304125) | Cod sursa (job #2706526)
#include <iostream>
#include <fstream>
#include <vector>
#include <cstring>
#include <set>
using namespace std;
int n,p,from,to,cost,i,dist[50005],nod,m,teste,t,x[50005],k;
vector < pair < int, int> > G[50005];
set < pair < int, int > > h;
ifstream in("distante.in");
ofstream out("distante.out");
int main()
{
in>>teste;
for (t=1; t<=teste; t++)
{
for (int i = 1; i <= n; i++)
G[i].clear();
in>>n>>m>>p;
for (i=1; i<=n; i++)
{
in>>x[i];
dist[i]=2e9;
}
for (i=1; i<=m; i++)
{
in>>from>>to>>cost;
G[from].push_back(make_pair(to,cost));
G[to].push_back(make_pair(from,cost));
}
dist[p]=0;
h.insert(make_pair(0,p));
while (!h.empty())
{
int nod=h.begin()->second;
h.erase(h.begin());
for (vector < pair < int, int > > ::iterator it=G[nod].begin(); it!=G[nod].end(); it++)
{
int to=it->first;
int cost=it->second;
if (dist[to]>dist[nod]+cost)
{
dist[to]=dist[nod]+cost;
h.insert(make_pair(dist[to],to));
}
}
}
k=0;
for (i=1; i<=n; i++)
{
if (dist[i]!=x[i])
{
k=1;
break;
}
}
if (k==1)
out<<"NU"<<endl;
else
out<<"DA"<<endl;
}
}