Cod sursa(job #1120303)

Utilizator lianaliana tucar liana Data 24 februarie 2014 23:00:29
Problema Distante Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.23 kb
#include<stdio.h>
#include<vector>
using namespace std;
#define nmax 50005
struct element{int n,c;};
int n, m, s, a, b, c, i, nrg, ii, t;
vector <element> ma[nmax];
vector <element> ::iterator it;
element el;
int dmin[nmax];
bool g[nmax], ok;

void citire()
{
    scanf("%ld %ld %ld",&n,&m,&s);
    for (i=1;i<=n;i++)
    {
        scanf("%ld",&dmin[i]);
        g[i]=0; ma[i].clear();
    }
    for(i=1;i<=m;i++)
    {
        scanf("%ld %ld %ld",&a,&b,&c);
        el.n=b; el.c=c;
        ma[a].push_back(el);
        el.n=a; el.c=c;
        ma[b].push_back(el);
    }
}

void rezolvare()
{
    ok=(dmin[s]==0);    nrg=0;
    for (i=1;((i<=n)&&(ok));i++)
        for (it=ma[i].begin();((it!=ma[i].end())&&(ok));it++)
        {
            if (dmin[(*it).n]==dmin[i]+(*it).c)
            {   nrg+=1-g[(*it).n];  g[(*it).n]=1;   }
            if (dmin[(*it).n]>dmin[i]+(*it).c)
                ok=0;
        }
    if ((ok)&&(nrg==(n-1)))
        printf("DA\n");
    else
        printf("NU\n");

}
int main()
{
    freopen("distante.in","r",stdin);
    freopen("distante.out","w",stdout);
    scanf("%ld",&t);
    for (ii=1;ii<=t;ii++)
    {
        citire();
        rezolvare();
    }
    return 0;
}