Cod sursa(job #2654278)

Utilizator AlexPascu007Pascu Ionut Alexandru AlexPascu007 Data 30 septembrie 2020 13:35:20
Problema Distante Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.54 kb
#include <bits/stdc++.h>
#define inf 2000000000
#define DIM 50010
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
int t,n,m,i,d[DIM],D[DIM],s,nod,vecin,cost,x,y,c;
set< pair<int,int> > S;
vector< pair<int,int> > L[DIM];
int dijkstra(int s) {
    for (int i=1;i<=n;i++)
        d[i]=inf;
    d[s]=0;
    S.insert({0,s});
    while (!S.empty()) {
        nod=S.begin()->second;
        S.erase(S.begin());
        for (auto it:L[nod]) {
            vecin=it.first, cost=it.second;
            if (d[vecin]>d[nod]+cost) {
                S.erase({d[vecin],vecin});
                d[vecin]=d[nod]+cost;
                if (d[vecin]<D[vecin])
                    return 0;
                S.insert({d[vecin],vecin});
            }
        }
    }
    return 1;
}
int main() {
    fin>>t;
    while (t--) {
        fin>>n>>m>>s;
        for (i=1;i<=n;i++)
            fin>>D[i];
        /*if (D[s]!=0) {
            fout<<"NU\n";
            continue;
        }*/
        for (i=1;i<=n;i++)
            L[i].clear();
        S.clear();
        for (i=1;i<=m;i++) {
            fin>>x>>y>>c;
            L[x].push_back({y,c});
            L[y].push_back({x,c});
        }
        if (!dijkstra(s)) {
            fout<<"NU\n";
            continue;
        }
        int ok=1;
        for (i=1;i<=n;i++)
            if (d[i]!=D[i]) {
                fout<<"NU\n";
                ok=0;
                break;
            }
        if (ok)
            fout<<"DA\n";
    }
    return 0;
}