Pagini recente » Cod sursa (job #489364) | Monitorul de evaluare | Cod sursa (job #2003985) | Monitorul de evaluare | Cod sursa (job #781625)
Cod sursa(job #781625)
#include <cassert>
#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
#define FORIT(it, v) for (typeof((v).begin()) it = (v).begin(); it != (v).end(); ++it)
const int N = 50005;
int t, n, m, s;
int dist[N];
vector <pair <int, int> > vec[N];
int verifica()
{
if (dist[s] != 0)
return 0;
for (int i = 1; i <= n; ++i) {
if (i == s)
continue;
bool ok = false;
FORIT(it, vec[i]) {
if (dist[i] + it->second < dist[it->first])
return 0;
if (dist[it->first] + it->second == dist[i])
ok = true;
}
if (!ok)
return 0;
}
return 1;
}
int main()
{
assert(freopen("distante.in", "r", stdin) != NULL);
assert(freopen("distante.out", "w", stdout) != NULL);
assert(scanf("%d", &t) == 1);
while (t--) {
assert(scanf("%d %d %d", &n, &m, &s) == 3);
for (int i = 1; i <= n; ++i)
assert(scanf("%d", &dist[i]) == 1);
for (int i = 1; i <= n; ++i) {
int x, y, z;
assert(scanf("%d %d %d", &x, &y, &z) == 3);
vec[x].push_back(make_pair(y, z));
vec[y].push_back(make_pair(x, z));
}
if (verifica())
printf("DA\n");
else
printf("NU\n");
}
}