Cod sursa(job #1855657)

Utilizator eduardturtoiEduard Turtoi eduardturtoi Data 23 ianuarie 2017 20:33:52
Problema Distante Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.66 kb
#include <fstream>
#include <utility>
#include <vector>
#include <set>
#define inf 0x3f3f3f3f
#define nmax 50001
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
vector<pair<int,int> >::iterator it;
vector<pair<int,int> >a[nmax];
set<pair<int,int> >heap;
int cost[nmax],d[nmax];
int n,m,t,start;

void citire()
{
    int i,x,y,c;
    fin>>n>>m>>start;
    for(i=1;i<=n;i++)
        fin>>cost[i];
    for(i=1;i<=m;i++)
    {
        fin>>x>>y>>c;
        a[x].push_back(make_pair(y,c));
        a[y].push_back(make_pair(x,c));
    }
}

void dijkstra()
{
    int i,nod,c,next;
    for(i=1;i<=n;i++)
        d[i]=inf;
    d[start]=0;
    heap.insert(make_pair(0,start));
    while(!heap.empty())
    {
        c=heap.begin()->first;
        nod=heap.begin()->second;
        heap.erase(heap.begin());
        for(it=a[nod].begin();it!=a[nod].end();it++)
        {
            next=it->first;
            c=it->second;
            if(d[next]>d[nod]+c)
            {
                if(!d[next])
                    heap.erase(heap.find(make_pair(d[next],next)));
                d[next]=d[nod]+c;
                heap.insert(make_pair(d[next],next));
            }
        }
    }
}

bool verif()
{
    int i;
    for(i=1;i<=n;i++)
        if(d[i]!=cost[i])
            return false;
    return true;
}

int main()
{
    int i,j;
    fin>>t;
    for(i=1;i<=t;i++)
    {
        citire();
        dijkstra();
        if(verif()==true)
            fout<<"DA";
        else
            fout<<"NU";
        fout<<'\n';
        for(j=1;j<=n;j++)
            a[j].clear();
    }
    return 0;
}