Cod sursa(job #2110878)

Utilizator BaldurCronos Baldur 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;
}