Cod sursa(job #2970222)

Utilizator MihaiCostacheCostache Mihai MihaiCostache Data 24 ianuarie 2023 17:24:08
Problema Distante Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.42 kb
#include <bits/stdc++.h>
#define INF 2000000005
using namespace std;

ifstream fin("distante.in");
ofstream fout("distante.out");
int t, n, m, sursa, x, y, c, dist[50001], d[50001];
vector <pair <int, int> > l[50001];

int main()
{
    fin>>t;
    while(t--)
    {
        fin>>n>>m>>sursa;
        for(int i=1; i<=n; i++)
        {
            fin>>dist[i];
        }
        for(int i=1; i<=m; i++)
        {
            fin>>x>>y>>c;
            l[x].push_back({y, c});
            l[y].push_back({x, c});
        }
        set <pair <int, int > > s;
        for(int i=1; i<=n; i++)
        {
            d[i]=INF;
        }
        d[sursa]=0;
        s.insert({0, sursa});
        while(!s.empty())
        {
            int nod=s.begin()->second;
            s.erase(s.begin());
            for(int i=0; i<l[nod].size(); i++)
            {
                int vecin=l[nod][i].first;
                int cost=l[nod][i].second;
                if(d[vecin]>d[nod]+cost)
                {
                    s.erase({d[vecin], vecin});
                    d[vecin]=d[nod]+cost;
                    s.insert({d[vecin], vecin});
                }
            }
        }
        int ok=1;
        for(int i=1; i<=n; i++)
        {
            l[i].clear();
            if(dist[i]!=d[i])
            {
                ok=0;
            }
        }
        if(ok==1)
            fout<<"DA\n";
        else
            fout<<"NU\n";
    }
    return 0;
}