Cod sursa(job #2858121)

Utilizator ValiAntonieAntonie Valentin ValiAntonie Data 27 februarie 2022 00:50:41
Problema Distante Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.26 kb
#include <bits/stdc++.h>
using namespace std;

ifstream fin("distante.in");
ofstream fout("distante.out");

int n,m,i,dist[500005],a,b,val,nod,vecin,viz[500005],sol[500005],t,ok,s;
priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>>Q;
vector <pair<int,int>> v[50005];

int main()
{
//lungime 0 nu infinit
fin>>t;
while(t--){
ok = 0;
fin>>n>>m>>s;
for(i=1;i<=n;i++){
    fin>>sol[i];
}
for(i=1;i<=m;i++){
    fin>>a>>b>>val;
    v[a].push_back({b,val});
    v[b].push_back({a,val});
}
dist[s] = 0;
//invers coada
Q.push({0,s});
for(i=1;i<=n;i++){
    if(i != s)
       dist[i] = 200000000;
    else
        dist[i] = 0;
    viz[i] = 0;
}
while(!Q.empty()){
    nod = Q.top().second;
    val = Q.top().first;
    Q.pop();
    if(!viz[nod]){
    for(i=0;i<v[nod].size();i++){
        vecin = v[nod][i].first;
        if(dist[vecin] > val + v[nod][i].second){
            dist[vecin] = val + v[nod][i].second;
            Q.push({dist[vecin],vecin});
        }

    }
    viz[nod] = 1;
    }
}
for(i=2;i<=n;i++){
    if(dist[i] == 200000000)
    dist[i] = 0;

    if(dist[i] != sol[i]){
        ok = 1;
        fout << "NU" << '\n';
        break;
    }
}
if(!ok)
    fout << "DA" << '\n';
}
    return 0;
}