Pagini recente » Cod sursa (job #1117012) | Cod sursa (job #253623) | Cod sursa (job #65081) | Cod sursa (job #314427) | Cod sursa (job #3176510)
#include <bits/stdc++.h>
#define oo 2e9
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
int n, m, k, c, T;
vector< pair <int, int > >L[50005];
priority_queue< pair <int, int > >q;
int d[50005], a[50005];
bitset<50005>viz;
void Reset()
{
viz.reset();
for(int i = 1; i <= n; i++)
L[i].clear();
}
void Dijkstra(int k)
{
int i, cost;
viz.reset();
for(int i = 1; i <= n; i++)
d[i] = oo;
q.push({0, k});
d[k] = 0;
while(!q.empty())
{
k = q.top().second;
q.pop();
if(viz[k] == 0)
{
viz[k] = 1;
for(auto e : L[k])
{
i = e.second;
cost = e.first;
if(d[i] > d[k] + cost)
{
d[i] = d[k] + cost;
q.push({-d[i], i});
}
}
}
}
}
int Valid(int a[100005])
{
for(int i = 1; i <= n; i++)
if(d[i] != a[i])return 0;
return 1;
}
void Citire()
{
int i, k, x, y, cnt = 0;
fin >> n >> m >> k;
for(i = 1; i <= n; i++)
{
fin >> a[i];
}
for(i = 1; i <= m; i++)
{
fin >> x >> y >> c;
L[x].push_back({c, y});
L[y].push_back({c, x});
}
Dijkstra(k);
if(Valid(a) == 1)fout << "DA" << "\n";
else fout << "NU";
}
int main()
{
fin >> T;
while(T--)
{
Citire();
Reset();
}
return 0;
}