Cod sursa(job #993245)

Utilizator andrettiAndretti Naiden andretti Data 3 septembrie 2013 15:40:29
Problema Distante Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.86 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 d[maxn],dc[maxn],used[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;
}

int check()
{
    queue <int> q;
    int n,m,s,k;
    int x,y,c;
    n=get_number(); m=get_number(); s=get_number();
    for(int i=1;i<=n;i++) d[i]=get_number(),used[i]=0,l[i].clear(),dc[i]=inf;

    if(d[s]) return 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])
         return 0;
        l[x].pb(mp(y,c));
        l[y].pb(mp(x,c));
    }

    while(!q.empty()) q.pop();
    for(q.push(s),dc[s]=0,used[s]=1;!q.empty();q.pop())
    {
        k=q.front();
        for(unsigned int i=0;i<l[k].size();i++)
         if(dc[k]+l[k][i].second<dc[l[k][i].first])
         {
             dc[l[k][i].first]=dc[k]+l[k][i].second;
             if(!used[l[k][i].first])
             {
                 q.push(l[k][i].first);
                 used[l[k][i].first]=1;
             }
         }
        used[k]=0;
    }
    for(int i=1;i<=n;i++)
     if(d[i]!=dc[i]) return 0;

    return 1;
}

void read()
{
    int t;
    scanf("%d\n",&t);
    for(int i=1;i<=t;i++)
     if(check()) 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;
}