Cod sursa(job #2110861)

Utilizator BaldurCronos Baldur Data 21 ianuarie 2018 14:23:43
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.4 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--) {
                v.reset();
                in >> n >> m >> s;

                for (int i = 1; i <= n; i++)
                        in >> d[i];

                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]) {
                                out << "NU\n";
                                correct = false;
                                break;
                        } else if (d[a] + c == d[b]) {
                                v[b] = 1;
                        }
                }

                if (!correct)
                        continue;

                for (int i = 1; i <= n; i++) {
                        if (!v[i] && i != s) {
                                correct = false;
                                break;
                        }
                }

                if (correct)
                        out << "DA\n";
        }
        return 0;
}