Cod sursa(job #1771468)

Utilizator tziplea_stefanTiplea Stefan tziplea_stefan Data 5 octombrie 2016 17:52:00
Problema Distante Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.73 kb
#include <fstream>
#include <vector>
#include <queue>
#define VAL 50005

using namespace std;

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

int T, N, M, i, j;
int a, b, c, S, k;
int cost[VAL], x[VAL];
vector<int> v[VAL];
vector<int> p[VAL];
queue<int> q;
bool ok;

void bfs()
{
    q.push(S);
    while (q.empty()==false)
    {
        a=q.front();
        q.pop();
        for (i=0; i<v[a].size(); i++)
        {
            if (cost[v[a][i]]==-1)
            {
                cost[v[a][i]]=cost[a]+p[a][i];
                q.push(v[a][i]);
            }
            else
            {
                if (cost[v[a][i]]>cost[a]+p[a][i])
                {
                    cost[v[a][i]]=cost[a]+p[a][i];
                    q.push(v[a][i]);
                }
            }
        }
    }
}

int main()
{
    fin >> T;
    for (k=1; k<=T; k++)
    {
        fin >> N >> M >> S;
        ok=true;
        for (i=1; i<=N; i++)
          fin >> x[i];
        for (i=1; i<=M; i++)
        {
            fin >> a >> b >> c;
            v[a].push_back(b);
            v[b].push_back(a);
            p[a].push_back(c);
            p[b].push_back(c);
        }
        if (x[S]!=0)
        {
            ok=false;
            fout << "NU\n";
        }
        else
        {
            for (i=1; i<=N; i++)
              if (i!=S)
                cost[i]=-1;
            bfs();
            for (i=1; i<=N; i++)
            {
                if (cost[i]!=x[i])
                  ok=false;
                cost[i]=0;
                v[i].clear();
                p[i].clear();
            }
            if (ok==true)
              fout << "DA\n";
            else
              fout << "NU\n";
        }
    }
    fin.close();
    fout.close();
    return 0;
}