Cod sursa(job #869339)

Utilizator ucnahHancu Andrei ucnah Data 1 februarie 2013 14:25:52
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.52 kb
#include <cstdio>
#include <vector>
#include <queue>

#define N 50001
#define infinit 999999999

using namespace std;

struct nod
{
    int v;
    int c;
};
int n,m,t,s;
int d[N],rezultat[N];
struct cmp
{
    bool operator()(const int &a, const int &b)const
    {
        return d[a]>d[b];
    }
};
vector<nod>h[N];
priority_queue<int,vector<int>,cmp>q;
void citire()
{

        scanf("%d %d %d",&n,&m,&s);
        for(int i=1;i<=n;i++)
            scanf("%d",&rezultat[i]);
        for(int i=1;i<=m;i++)
        {
            int v;
            nod a;
            scanf("%d %d %d",&v,&a.v,&a.c);
            h[v].push_back(a);
        }
}
void dijkstra()
{
    d[s]=0;
    for(int i=1;i<=n;i++)
    {
        if(i!=s)
            d[i]=infinit;
    }
    q.push(s);
    while(!q.empty())
    {
        int v=q.top();
        q.pop();
        int nr=h[v].size();
        for(int i=0;i<nr;i++)
        {
            nod p=h[v][i];
            if(d[v]+p.c<d[p.v])
            {
                d[p.v]=d[v]+p.c;
                q.push(p.v);
            }
        }
    }
}
void afisare()
{
    for(int i=1;i<=n;i++)
    {
        if(d[i]!=rezultat[i])
        {
            printf("NU\n");
            return;
        }
    }
    printf("DA\n");
}
int main()
{
    freopen("distante.in","r",stdin);
    freopen("distante.out","w",stdout);
    scanf("%d",&t);
    for(int i=0;i<t;i++)
    {
        citire();
        dijkstra();
        afisare();
    }
    return 0;
}