Cod sursa(job #857484)

Utilizator marta_diannaFII Filimon Marta Diana marta_dianna Data 17 ianuarie 2013 21:12:51
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.24 kb
#include<fstream>
#include<vector>
#include<queue>
#include<cstdlib>
#define NMAX 50010
#define INF 100000010

using namespace std;

ifstream f("distante.in");
ofstream g("distante.out");

struct muchie
{
    int nod, cost;
};

int T, n, m, S, d[NMAX], sol[NMAX];
vector<muchie> a[NMAX];
queue<int> q;

void Citeste()
{
    int i, x, y;
    muchie z;
    char c[40];

    f>>n>>m>>S;

    for (i=1; i<=n; ++i)
    {
        f>>sol[i];
        a[i].clear();
        d[i]=INF;
    }

    for (i=1; i<=m; ++i)
    {
        f>>x>>y>>z.cost;
        z.nod=y; a[x].push_back(z);
        z.nod=x; a[y].push_back(z);
    }
}

void Solve()
{
    int i, x, lg, nr=0;
    muchie z;

    d[S]=0; q.push(S);

    if (sol[S]==0)

    while (!q.empty())
    {
        x=q.front(); q.pop();

        for (i=0; i<a[x].size(); ++i)
        {
            z=a[x][i];
            lg=z.cost+d[x];
            if (lg==sol[z.nod] && d[x]!=sol[z.nod])
            {
                d[z.nod]=lg; ++nr;
                q.push(z.nod);
            }
        }
    }

    if (nr==n) g<<"DA\n";
    else g<<"NU\n";
}

int main()
{
    f>>T;

    while (T--)
    {
//        Initializeaza();
        Citeste();
        Solve();
//        Scrie();
    }

    f.close();
    g.close();
    return 0;
}