Pagini recente » Cod sursa (job #2300921) | Cod sursa (job #2469732) | Cod sursa (job #1518314) | Cod sursa (job #1390232) | Cod sursa (job #2915364)
#include <fstream>
#include <queue>
#include <vector>
#include <climits>
#pragma GCC optimize("O1")
#pragma GCC optimize("O2")
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
using namespace std;
ifstream in ("distante.in");
ofstream out ("distante.out");
const int max_size = 5e4 + 1, INF = 1e9 + 1;
int d[max_size], viz[max_size], rez[max_size];
vector <pair <int, int>> edges[max_size];
struct cmp{
bool operator()(int x, int y)
{
return d[x] > d[y];
}
};
priority_queue <int, vector <int>, cmp> pq;
void djk (int nod)
{
viz[nod] = 1;
d[nod] = 0;
pq.push(nod);
while (!pq.empty())
{
nod = pq.top();
pq.pop();
viz[nod] = 0;
for (auto f : edges[nod])
{
if (d[nod] + f.second < d[f.first])
{
d[f.first] = d[nod] + f.second;
if (viz[f.first] == 0)
{
viz[f.first] = 1;
pq.push(f.first);
}
}
}
}
}
void solve ()
{
int n, m, x;
in >> n >> m >> x;
for (int i = 1; i <= n; i++)
{
edges[i].clear();
in >> rez[i];
d[i] = INF;
}
while (m--)
{
int a, b, c;
in >> a >> b >> c;
edges[a].push_back({b, c});
edges[b].push_back({a, c});
}
djk(x);
bool flag = true;
for (int i = 1; i <= n; i++)
{
if (d[i] != rez[i])
{
flag = false;
break;
}
}
if (flag)
{
out << "DA";
}
else
{
out << "NU";
}
out << '\n';
}
int main ()
{
ios_base::sync_with_stdio(false);
//in.tie(0);
//out.tie(0);
int t;
in >> t;
while (t--)
{
solve();
}
in.close();
out.close();
return 0;
}