Cod sursa(job #888684)

Utilizator narcis_c2007Ciobotariu Narcis Paul Dumitru narcis_c2007 Data 24 februarie 2013 11:58:07
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.62 kb
#include <iostream>
#include<fstream>
const int inf=1<<30;
using namespace std;
struct graf
{
    int cost,nod;
    graf *next;
};
graf *a[50001];
int n,m,verif[50001],d[50001],dist[50001],s;
ifstream f("distante.in");
void add(int unde,int ce,int cost)
{
    graf *q=new graf;
    q->cost=cost;
    q->nod=ce;
    q->next=a[unde];
    a[unde]=q;
}
void citire()
{
    int i,x,y,z;

    for(i=1;i<=m;i++)
    {
        f>>x>>y>>z;
        add(x,y,z);
    }
}
void dijkstra(int inceput)
{
    int i,j,min,pmin=0;
    for(i=1;i<=n;i++)
        if(i!=inceput)
            d[i]=inf;
        d[inceput]=0;
    for(i=1;i<=n;i++)
    {
        min=inf;
        for(j=1;j<=n;j++)
            if(min>d[j]&&verif[j]==0)
        {
            min=d[j];
            pmin=j;
        }
        verif[pmin]=1;
        graf *t=a[pmin];
        while(t)
        {
            if(d[t->nod]>d[pmin]+t->cost)
                d[t->nod]=d[pmin]+t->cost;
            t=t->next;
        }

    }
    a[inceput]->next=NULL;
}
int main()
{
    int t,r,ok;
    f>>t;
    ofstream g("distante.out");
    for(int i=1;i<=t;i++)
    {

        f>>n>>m>>s;
        for(int j=1;j<=n;j++)
            f>>dist[j];
        citire();
        dijkstra(s);
        for(int j=1;j<=n;j++)
            verif[j]=0;
             ok=0;
        for(int j=1;j<=n;j++)
        {
            if(d[j]!=inf)
            r=d[j];
        else
            r=0;

        if(r!=dist[j])
            ok=1;
        }
        if(ok==1)
            g<<"NU"<<endl;
        else
            g<<"DA"<<endl;

    }
    return 0;
}