Cod sursa(job #993278)

Utilizator andrettiAndretti Naiden andretti Data 3 septembrie 2013 16:24:25
Problema Distante Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.7 kb
#include<stdio.h>
#include<vector>
#include<queue>
#define pb push_back
#define mp make_pair
#define maxn 50005
#define maxb 8192
#define inf 0x3f3f3f3f
using namespace std;

int t,n,m,s;
int x,y,c,ok;
int d[maxn],v[maxn];
vector < pair<int,int> > l[maxn];
char buff[maxb];
int pos=0;

int get_number()
{
    int nr=0;
    while(buff[pos]<'0' || buff[pos]>'9')
     if(++pos==maxb) fread(buff,1,maxb,stdin),pos=0;

    while(buff[pos]>='0' &&  buff[pos]<='9')
    {
        nr=nr*10+buff[pos]-'0';
        if(++pos==maxb) fread(buff,1,maxb,stdin),pos=0;
    }
    return nr;
}

void read()
{
    queue <int> q;
    int k;

    t=get_number();
    for(int it=1;it<=t;it++)
    {
        ok=1;
        n=get_number(); m=get_number(); s=get_number();
        for(int i=1;i<=n;i++) d[i]=get_number(),v[i]=0,l[i].clear();

        if(d[s]) ok=0;
        for(int i=1;i<=m;i++)
        {
            x=get_number(); y=get_number(); c=get_number();
            if(d[x]+c<d[y] || d[y]+c<d[x]) ok=0;
            l[x].pb(mp(y,c));
            l[y].pb(mp(x,c));
        }
        while(!q.empty()) q.pop();
        for(q.push(s),v[s]=1;!q.empty();q.pop())
        {
            k=q.front();
            for(unsigned int i=0;i<l[k].size();i++)
             if(!v[l[k][i].first] && d[k]+l[k][i].second==d[l[k][i].first])
                 q.push(l[k][i].first),v[l[k][i].first]=1;
        }
        for(int i=1;i<=n;i++)
         if(!v[i]) ok=0;

        if(ok) printf("DA\n");
        else printf("NU\n");
    }
}

int main()
{
    freopen("distante.in","r",stdin);
    freopen("distante.out","w",stdout);

    read();
    fclose(stdin);
    fclose(stdout);
    return 0;
}