Cod sursa(job #2110878)
Utilizator | Data | 21 ianuarie 2018 14:36:40 | |
---|---|---|---|
Problema | Distante | Scor | 40 |
Compilator | cpp | Status | done |
Runda | Arhiva de probleme | Marime | 1.63 kb |
#include <iostream>
#include <fstream>
#include <bitset>
#define N 50002
using namespace std;
ifstream in("distante.in");
ofstream out("distante.out");
int main() {
bitset<N> v;
int t, n, m, s, a, b, c, d[N];
in >> t;
while (t--) {
in >> n >> m >> s;
for (int i = 1; i <= n; i++) {
in >> d[i];
v[i] = 0;
}
v[s] = 1;
if (d[s] != 0) {
out << "NU\n";
continue;
}
bool correct = true;
for (int i = 1; i <= m; i++) {
in >> a >> b >> c;
if (d[a] + c < d[b] || d[b] + c < d[a]) {
correct = false;
break;
} else if (d[a] + c == d[b]) {
v[b] = 1;
} else if (d[b] + c == d[a]) {
v[a] = 1;
}
}
if (!correct) {
out << "NU\n";
continue;
}
for (int i = 1; i <= n; i++) {
if (!v[i]) {
correct = false;
break;
}
}
if (correct)
out << "DA\n";
else
out << "NU\n";
}
return 0;
}