Cod sursa(job #2771597)

Utilizator Raresr14Rosca Rares Raresr14 Data 28 august 2021 12:10:18
Problema Distante Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.32 kb
#include <bits/stdc++.h>
#define X first
#define Y second
#define INF 100000000
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
int n,m,start,i,x,y,z,D[50010],D1[50010],ok,t,nod;
set<pair<int,int> > s;
vector<pair<int,int> > L[50010];
int main(){
    for(fin>>t;t--;){
        fin>>n>>m>>start;
        for(i=1;i<=n;i++)
            fin>>D1[i],D[i]=INF;
        for(i=1;i<=m;i++){
            fin>>x>>y>>z;
            L[x].push_back(make_pair(y,z));
            L[y].push_back(make_pair(x,z));
        }
        D[start]=0;
        s.insert(make_pair(0,start));
        ok=1;
        while(!s.empty()){
            nod=s.begin()->Y;
            s.erase(s.begin());
            for(i=0;i<L[nod].size();i++){
                int vec=L[nod][i].X;
                int cost=L[nod][i].Y;
                if(D[vec]>D[nod]+cost){
                    s.erase(make_pair(D[vec],vec));
                    D[vec]=D[nod]+cost;
                    s.insert(make_pair(D[vec],vec));
                }
            }
        }
        for(i=1;i<=n;i++){
            L[i].clear();
            if(D1[i]!=D[i]){
                ok=0;
                break;
            }
        }
        if(ok)
            fout<<"DA\n";
        else
            fout<<"NU\n";
    }
    return 0;
}