Cod sursa(job #3271750)

Utilizator GILIEDAVIDGilie David Florin GILIEDAVID Data 27 ianuarie 2025 10:16:56
Problema Distante Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.49 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
#define per pair<int,int>
#define inf 2e9
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
priority_queue<per,vector<per>,greater<per> > h;
vector<per> v[50005];
int d[50005],i,n,m,x,y,z,a[50005],ok,p,k,j;
void dijkstra(int k)
{
    int i,t,u,c;
    for(i=1;i<=n;i++)
        d[i]=inf;
    d[k]=0;
    h.push({0,k});
    while(!h.empty())
    {
        t=h.top().second;
        c=h.top().first;
        h.pop();
        if(d[t]!=c) continue;
        for(i=0;i<v[t].size();i++)
        {
            u=v[t][i].second;
            c=v[t][i].first;
            if(d[t]+c<d[u])
            {
                d[u]=d[t]+c;
                h.push({d[u],u});
            }
        }
    }
}
int main()
{
    f>>p;
    for(i=1;i<=p;i++)
        {
            f>>n>>m>>k;
            for(j=1;j<=n;j++)
                f>>a[j];
            for(j=1;j<=m;j++)
            {
                f>>x>>y>>z;
                v[x].push_back({z,y});
                v[y].push_back({z,x});
            }
            dijkstra(k);
            ok=1;
            for(j=1;j<=n;j++)
            {
                if(a[j]!=d[j])
                {
                    ok=0;
                    break;
                }
            }
            if(ok) g<<"DA"<<'\n';
            else g<<"NU"<<'\n';
            for(j=1;j<=n;j++)
                v[j].clear();
        }
    return 0;
}