Cod sursa(job #2110904)

Utilizator BaldurCronos Baldur Data 21 ianuarie 2018 14:52:18
Problema Distante Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.15 kb
#include <iostream>
#include <fstream>
#include <bitset>
#define N 50002
using namespace std;
ifstream in("distante.in");
ofstream out("distante.out");
bitset<N> v;
int t, n, m, s, a, b, c, d[N];

bool check() {
        if (d[s])
                return false;
        v[s] = 1;

        for (int i = 1; i <= m; i++) {
                in >> a >> b >> c;
                if (d[a] + c < d[b] || d[b] + c < d[a])
                        return false;

                if (d[a] + c == d[b])
                        v[b] = 1;

                if (d[b] + c == d[a])
                        v[a] = 1;
        }

        for (int i = 1; i <= n; i++) {
                if (!v[i])
                        return false;
        }

        return true;
}

int main() {
        in >> t;

        while (t--) {
                in >> n >> m >> s;
                for (int i = 1; i <= n; i++) {
                        in >> d[i];
                        v[i] = 0;
                }

                if (check())
                        out << "DA\n";
                else
                        out << "NU\n";
        }
        return 0;
}