Cod sursa(job #1089983)

Utilizator pulseOvidiu Giorgi pulse Data 22 ianuarie 2014 09:52:38
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.55 kb
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>
#define INF 0x3f3f3f3f
#define NMAX 50002

using namespace std;

ifstream fin ("distante.in"); ofstream fout ("distante.out");

int T, N, M, cost[NMAX], start, dZ[NMAX];
vector <int> v[NMAX];
queue <int> q;

void ReadData ()
{
    fin >> N >> M >> start;
    for (int i = 1; i <= N; ++i)
        fin >> dZ[i];
    for (int i = 1, a, b, c; i <= M; ++i)
    {
        fin >> a >> b >> c;
        v[a].push_back (b);
        v[a].push_back (c);
        v[b].push_back (a);
        v[b].push_back (c);
    }
    for (int i = 1; i <= N; ++i)
    {
        if (i != start)
            cost[i] = INF;
    }
}

void Solve ()
{
    q.push (start);
    while (!q.empty ())
    {
        int x = q.front ();
        for (int i = 0; i < v[x].size (); i += 2)
        {
            if (cost[v[x][i]] > cost[x] + v[x][i + 1])
            {
                q.push (v[x][i]);
                cost[v[x][i]] = cost[x] + v[x][i + 1];
            }
        }
        q.pop ();
    }
}

int Check ()
{
    cost[start] = 0;
    for (int i = 1; i <= N; ++i)
    {
        if (cost[i] != INF)
        {
            if (cost[i] != dZ[i]) return 0;
        }
        else return 0;
    }
    return 1;
}

int main ()
{
    fin >> T;
    for (int i = 1; i <= T; ++i)
    {
        ReadData ();
        Solve ();
        if (Check ())
            fout << "DA\n";
        else
            fout << "NU\n";
    }
    fin.close (); fout.close ();
    return 0;
}