Cod sursa(job #1770457)

Utilizator Andrei1998Andrei Constantinescu Andrei1998 Data 4 octombrie 2016 14:19:17
Problema Distante Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.65 kb
#include <iostream>
#include <fstream>
#include <utility>

using namespace std;

class InputReader {
    public:
        InputReader() {}
        InputReader(const char *file_name) {
            input_file = fopen(file_name, "r");
            cursor = 0;
            fread(buffer, SIZE, 1, input_file);
        }
        inline InputReader &operator >>(int &n) {
            while(buffer[cursor] < '0' || buffer[cursor] > '9') {
                advance();
            }
            n = 0;
            while('0' <= buffer[cursor] && buffer[cursor] <= '9') {
                n = n * 10 + buffer[cursor] - '0';
                advance();
            }
            return *this;
        }
    private:
        FILE *input_file;
        static const int SIZE = 1 << 17;
        int cursor;
        char buffer[SIZE];
        inline void advance() {
            ++ cursor;
            if(cursor == SIZE) {
                cursor = 0;
                fread(buffer, SIZE, 1, input_file);
            }
        }

    };


const int NMAX = 50000 + 5;
const int MMAX = 2 * NMAX;

int d[NMAX];

InputReader _cin("distante.in");

bool test() {
    int n, m, s;
    _cin >> n >> m >> s;

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

    bool fail = false;
    if (d[s] != 0)
        fail = true;
    for (int i = 1; i <= m; ++ i) {
        int a, b, c;
        _cin >> a >> b >> c;

        if (d[a] + c < d[b] || d[b] + c < d[a])
            fail = true;
    }

    return !fail;
}

int main()
{
    freopen("distante.out", "w", stdout);

    int t;
    _cin >> t;

    while (t --)
        if (test())
            cout << "DA\n";
        else
            cout << "NU\n";

    return 0;
}