Cod sursa(job #2444205)

Utilizator Ionut28Porumb Palincas Ionut Ionut28 Data 30 iulie 2019 15:55:06
Problema Distante Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.13 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
const int nmax = 50005;
int n, m, g, s, d1[nmax], d[nmax];
int main()
{
    fin >> g;
    for(int t = 1; t <= g; ++t)
    {
        fin >> n >> m >> s;
        for(int i = 1; i <= n; ++i)
            fin >> d1[i];
        for(int i = 1; i <= m; ++i)
        {
            int x, y, c;
            fin >> x >> y >> c;
            if(x == s)
            {
                d[y] = c;
            }
            else if(y == s)
            {
                d[x] = c;
            }
            else
            {
               if(d[x] > 0 && d[y] == 0)
                d[y] = c + 1;
               else if(d[y] > 0 && d[x] == 0)
               d[x] = c + 1;

            }

        }
        d[s] = 0;
        bool ok = true;
        for(int i = 1; i <= n; ++i)
        {
            if(d[i] != d1[i])
            {
                ok = false;
                break;
            }
        }
        if(ok)
            fout << "DA\n";
        else
            fout << "NU\n";
    }
    return 0;
}