Cod sursa(job #2771591)

Utilizator Raresr14Rosca Rares Raresr14 Data 28 august 2021 12:04:07
Problema Distante Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.31 kb
#include <bits/stdc++.h>
#define X first
#define Y second
#define INF 1000000
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[100010];
int main(){
    for(fin>>t;t--;){
        fin>>n>>m>>start;
        for(i=1;i<=n;i++)
            fin>>D1[i],D[i]=INF,L[i].clear();
        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()&&ok){
            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){
                    if(D[nod]+cost<D1[vec]){
                        ok=0;
                        break;
                    }
                    s.erase(make_pair(D[vec],vec));
                    D[vec]=D[nod]+cost;
                    s.insert(make_pair(D[vec],vec));
                }
            }
        }
        if(ok)
            cout<<"DA\n";
        else
            cout<<"NU\n";
    }
    return 0;
}