Cod sursa(job #1240078)

Utilizator ThomasFMI Suditu Thomas Thomas Data 10 octombrie 2014 14:34:31
Problema Distante Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <fstream>
#include <vector>
using namespace std;

#define NMax 50005

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

int T;
int n,m,s,i,ok;
int D[NMax],viz[NMax];
int a,b,c;

int main()
{
    f>>T;
    while(T--)
    {
        ok=1;

        f>>n>>m>>s;
        for(i=1;i<=n;++i)
        {
            f>>D[i];
            viz[i]=0;
        }

        if(D[s]!=0) ok=0;
        viz[s]=1;

        for(i=1;i<=m;++i)
        {
            f>>a>>b>>c;

            if(D[a]+c<D[b]) ok=0;
            if(D[b]+c<D[a]) ok=0;
            if(D[a]+c==D[b]) viz[b]=1;
            if(D[b]+c==D[a]) viz[a]=1;
        }

        for(i=1;i<=n && ok;++i) if(viz[i]==0) ok=0;

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

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