Cod sursa(job #2654740)

Utilizator RaduXD1Nicolae Radu RaduXD1 Data 2 octombrie 2020 10:17:25
Problema Distante Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <fstream>
#include <queue>
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
vector<pair <int, int> > v[50003];
priority_queue< pair<int, int> , vector<pair<int, int> >, greater<pair<int, int> > > q;
int n, m, i, a, b, c, cost, el , d[50003],s , t[50003],x,aux[50003];

void sterge()
{
    for(int i=1;i<=n;i++) v[i].clear();
}

int main()
{
    fin>>x;
    while(x>0)
    {
        fin>>n>>m>>s;
        for(i=1;i<=n;i++)
        {
            fin>>aux[i];
            d[i]=10000000;t[i]=0;
        }
        d[s]=0;
        for(i=1;i<=m;i++)
        {
            fin>>a>>b>>c;
            if(!q.empty()) q.pop();
            v[a].push_back(make_pair(b, c));
            v[b].push_back(make_pair(a, c));
        }
        for(auto it:v[s])
        {
            q.push(make_pair(it.second, it.first));
            t[it.first]=s;
        }
        int nr=n-1;
        while(nr>0)
        {
            cost=q.top().first;
            el=q.top().second;
            q.pop();
            if(d[el]>cost)
            {
                for(auto it:v[el]) if(t[el]!=it.first)
                {
                    q.push(make_pair(it.second+cost, it.first));
                    t[it.first]=el;
                }nr--;
                d[el]=cost;
            }
        }
        int ok=1;
        for(i=1;i<=n;i++)
        {
            if(aux[i]!=d[i]) ok=0;
            v[i].clear();
        }
        if(ok==0)
            fout<<"NU\n";
        else
            fout<<"DA\n";
        x--;
    }
    return 0;
}