Cod sursa(job #2103846)

Utilizator edynator34Nechitoaia George-Edward edynator34 Data 10 ianuarie 2018 20:49:20
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.64 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#define mx 50005
#define INF (1<<30)
using namespace std;
ifstream in("distante.in");
ofstream out("distante.out");
int DCO;
int verificare[mx];
bool v[mx];
int  d[mx];
vector < pair < int , int > > A[mx];
int x,y,c,i,n,m,s;

struct compara
{
    bool operator()(int x, int y){
            return d[x]>d[y];
    }
};
priority_queue < int ,vector< int > , compara > q;

void Dijkstra(int sursa){
    for(i=1;i<=n;i++){
        d[i]=INF;
    }
    d[sursa]=0;
    v[sursa]=1;
    q.push(sursa);
    vector < pair < int , int > >::iterator it;
    while (!q.empty()){
        int nodcurent=q.top();
        q.pop();
        v[nodcurent]=false;
        for(it=A[nodcurent].begin() ; it!=A[nodcurent].end() ; it++){
            int vecin=it->first;
            int cost=it->second;
            if(d[vecin]>d[nodcurent]+cost){
                d[vecin]=d[nodcurent]+cost;
                if(v[vecin]==false){
                    v[vecin]=1;
                    q.push(vecin);
                }
            }
        }

    }

}

int verificareF (){
    for(i=1;i<=n;++i){
        if(verificare[i] != d[i]) return 0;
    }
    return 1;
}
int main()
{
    in>>DCO;
    while(DCO){
        in>>n>>m>>s;
        for(i=1;i<=n;++i)
            {
                in>>verificare[i];
            }
        for(i=1;i<=m;++i){
            in>>x>>y>>c;
            A[x].push_back(make_pair(y,c));
        }
        Dijkstra(s);
        if(verificareF()) out<<"DA"<<'\n';
        else out<<"NU"<<'\n';


        DCO--;
    }
    return 0;
}