Cod sursa(job #2476762)

Utilizator iulianarsenoiuArsenoiu Iulian iulianarsenoiu Data 19 octombrie 2019 11:23:32
Problema Distante Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.5 kb
#include <bits/stdc++.h>

using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
typedef pair<int,int> graf;
vector <graf> G[50005];
priority_queue <graf,vector<graf>,greater<graf>> h;
int t,n,m,x,y,i,j,d[50005],dc[50005],pl,c;
bool sel[50005];
void dijkstra(int pl)
{
    for(int i=1;i<=n;i++)
    {
        d[i]=INT_MAX;
        sel[i]=false;
    }
    sel[pl]=true;
    d[pl]=0;
    for(auto it : G[pl])
    {
        h.push(it);
        d[it.second]=it.first;
    }
    while(!h.empty())
    {
        while(!h.empty() && sel[h.top().second])
            h.pop();
        if(h.empty())
            return;
        graf k=h.top();
        sel[k.second]=true;
        h.pop();
        for(auto it : G[k.second])
        {
            if(k.first+it.first<d[it.second])
            {
                d[it.second]=k.first+it.first;
                h.push({d[it.second],it.second});
            }
        }
    }
}
int main()
{
    f>>t;
    for(j=1;j<=t;j++)
    {
        f>>n>>m>>pl;
        for(i=1;i<=n;i++)
            f>>dc[i];
        for(i=1;i<=n;i++)
            G[i].clear();
        for(i=1;i<=m;i++)
        {
            f>>x>>y>>c;
            G[x].push_back({c,y});
            G[y].push_back({c,x});
        }
        dijkstra(pl);
        bool ok=true;
        for(i=1;i<=n;i++)
            if(d[i]!=dc[i])
            ok=false;
        if(ok==true)
            g<<"DA"<<'\n';
        else g<<"NU"<<'\n';
    }
    return 0;
}