Cod sursa(job #2583032)

Utilizator mirceaisherebina mircea mirceaishere Data 17 martie 2020 18:14:48
Problema Distante Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.42 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");

int teste, start;
int n, m, i, j, x, y, c, d[50010], dPresupus[50010], v[50010], nod, vecin, cost;
vector < pair<int, int> > a[50010];
queue <int> s;

bool bellman(){
    for(i=1; i<=n; i++){
        d[i]=1000000000;
        v[i]=0;
    }
    s.push(start);
    v[start]=1;
    d[start]=0;
    while(!s.empty()){
        nod=s.front();
        v[nod]=0;
        for(i=0; i<a[nod].size(); i++){
            vecin=a[nod][i].first;
            cost=a[nod][i].second;
            if(d[nod]+cost<d[vecin]){
                d[vecin]=d[nod]+cost;
                if(v[vecin]==0){
                    s.push(vecin);
                    v[vecin]=1;
                }
            }
        }
        s.pop();
    }
    for(i=1; i<=n; i++){
        if(d[i]!=dPresupus[i]){
            return 0;
        }
    }
    return 1;
}

int main(){
    fin>>teste;
    for(; teste; teste--){
        fin>>n>>m>>start;
        for(i=1; i<=n; i++){
            fin>>dPresupus[i];
            a[i].clear();
        }
        for(i=1; i<=m; i++){
            fin>>x>>y>>c;
            a[x].push_back({y, c});
            a[y].push_back({x, c});
        }
        if(bellman()==1){
            fout<<"DA"<<"\n";
        }else{
            fout<<"NU"<<"\n";
        }
    }
}