Pagini recente » Cod sursa (job #310107) | Cod sursa (job #1787674) | Cod sursa (job #2449226) | Cod sursa (job #463278) | Cod sursa (job #1315884)
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>
#define INF 0x3f3f3f3f
using namespace std;
ifstream is("distante.in");
ofstream os("distante.out");
int T, n, m, s;
int d[50001], a[50001];
vector<pair<int, int> > G[50001];
priority_queue<pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > > Q;
void Read();
void Dijkstra(int x);
bool Ok();
int main()
{
is >> T;
for ( int t = 0; t < T; ++t )
{
Read();
Dijkstra(s);
if ( Ok() )
os << "DA\n";
else
os << "NU\n";
}
is.close();
os.close();
return 0;
}
void Read()
{
is >> n >> m >> s;
for ( int i = 1; i <= n; ++i )
{
is >> a[i];
G[i].clear();
}
for ( int i = 0, x, y, c; i < m; ++i )
{
is >> x >> y >> c;
G[x].push_back({y, c});
G[y].push_back({x, c});
}
}
void Dijkstra(int x)
{
int c;
memset(d, 63, sizeof(d) );
d[x] = 0;
Q.push({0, x});
while ( !Q.empty() )
{
c = Q.top().first;
x = Q.top().second;
Q.pop();
for (const auto& v : G[x] )
if ( d[v.first] > c + v.second )
{
d[v.first] = c + v.second;
Q.push({d[v.first], v.first});
}
}
}
bool Ok()
{
for ( int i = 1; i <= n; ++i )
{
if ( d[i] == INF )
d[i] = 0;
if ( d[i] != a[i] )
return false;
}
return true;
}