Cod sursa(job #2060122)

Utilizator aeromaniaXRadoi Iulian aeromaniaX Data 7 noiembrie 2017 21:12:06
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.42 kb
#include <bits/stdc++.h>
using namespace std;


#define INF 999999999
queue <int> q;
vector <int> G[50001],C[50001];


int d1[50001],d2[50001],x,y,c,m,n,p,L,xi,yi;

void bfs(int xs,int d[])
{
    q.push(xs);
    d[xs]=0;

    while(!q.empty()){
        x = q.front();
        for(int i=0;i<G[x].size();i++){
        //    ap[p][G[x][i]]=1;
            if(d[G[x][i]] > d[x] + C[x][i])
        {
            d[G[x][i]] = d[x] + C[x][i];
            q.push(G[x][i]);


        }}
        q.pop();
    }
}

bool check()
{
    for(int i=1;i<=n;i++)
        if(d1[i]!=d2[i])
            return 0;
    return 1;
}
int main()
{
    freopen("distante.in","r",stdin);
    freopen("distante.out","w",stdout);
int t;
    scanf("%d",&t);
    while(t--)
    {
//        for(int i=0;i<G[i].size();i++)
//                G[i].clear();
//        for(int i=0;i<C[i].size();i++)
//                C[i].clear();


        scanf("%d %d %d",&n,&m,&p);
        for(int i=1;i<=n;i++)
            scanf("%d",&d2[i]);

        for(int i=1;i<=n;i++)
            d1[i]=INF;

        while(m--){
            scanf("%d %d %d",&x,&y,&c);

            G[x].push_back(y);
            C[x].push_back(c);
        }
        bfs(p,d1);

//        for(int i=1;i<=n;i++)
//            printf("%d ",d1[i]);
//        printf("\n");

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


   return 0;
}