Cod sursa(job #892535)

Utilizator vlcmodanModan Valentin vlcmodan Data 26 februarie 2013 10:27:15
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.47 kb

#include<fstream>

#define inf 1000000000
#define NMax 100003

using namespace std;

ifstream f("distante.in");
ofstream g("distante.out");

struct Nod
{
    int nod, cost;
    Nod *next;
};

Nod *t,*Vecin[NMax];

int j,m,n,x,y,costul,nod,d[NMax],coada[5*NMax];

void adauga(int x, int y, int costul)
{
    Nod *q = new Nod;
    q->nod = y;
    q->cost = costul;
    q->next = Vecin[x];
    Vecin[x] = q;
}

void bellman_ford(int k)
{
    int p,u,i;
    for(i=1;i<=n;i++)
        d[i]=inf;
    p=1;
    u=1;
    coada[u]=k;
    d[k]=0;
    while (p<=u)
     {
        j=coada[p];p++;
        t=Vecin[j];
        while(t)
        {
            if(d[t->nod]>d[j]+t->cost)
            {
                coada[++u]=t->nod;
                d[t->nod]=d[j]+t->cost;
            }

            t=t->next;
        }
    }
}

int main()
{
    int i,k,zaharel[10000],T,ok;
    f>>T;
    for(int j=1; j<=T; j++)
    {
        f>>n>>m>>k;
        for(i=1; i<=n; i++)
            f>>zaharel[i];
        for(i=1;i<=m;i++)
        {
            f>>x>>y>>costul;
            adauga(x,y,costul);
        }
        bellman_ford(k);
        ok=0;
        for ( int i = 1; i <= n; i++ )
            if((d[i] == inf && zaharel[i]==0)||(d[i]==zaharel[i]))
                ok=0;
            else
                ok=1;
        if(ok==0)
            g<<"DA\n";
        else
            g<<"NU\n";
    }
    f.close();
    g.close();
    return 0;
}