Cod sursa(job #2276864)

Utilizator ciutanpCiuta Andrei Calin ciutanp Data 5 noiembrie 2018 15:47:12
Problema Distante Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.61 kb
#include<bits/stdc++.h>
using namespace std;

ifstream f("distante.in");
ofstream g("distante.out");

int t,n,m,s,dr[50001],dn[50001];
vector< pair<int,int> >G[50001];
priority_queue< pair< int,int >, vector< pair<int,int> >,greater< pair<int,int> > >Q;
bool viz[50001];

void dij(int start)
{
    for(int i=1;i<=n;++i)
        dn[i]=INT_MAX;
    dn[start]=0;
    Q.push({0,start});
    while(!Q.empty())
    {
        int nodCurent=Q.top().second;
        Q.pop();
        if(viz[nodCurent])
            continue;
        viz[nodCurent]=1;

        for(int i=0;i<G[nodCurent].size();++i)
        {
            int cost=G[nodCurent][i].second;
            int nodveci=G[nodCurent][i].first;
            if(cost+dn[nodCurent]<dn[nodveci])
            {
                dn[nodveci]=cost+dn[nodCurent];
            }
            if(!viz[nodveci])
            {
                Q.push({dn[nodveci],nodveci});
            }
        }
    }
}

int main()
{
    f>>t;
    while(t--)
    {
        f>>n>>m>>s;
        memset(viz,0,sizeof(viz));
        for(int i=1;i<=n;++i)
            {
                f>>dr[i];
                G[i].clear();
            }
        for(int i=1;i<=m;++i)
        {
            int x,y,c;
            f>>x>>y>>c;
            G[x].push_back({y,c});
            G[y].push_back({x,c});
        }
        dij(s);
        bool ok=1;
        for(int i=1;i<=n && ok;++i)
        {
            if(dr[i]!=dn[i])
            {
                ok=0;
            }
        }
        if(ok==0)
            g<<"NU\n";
        else
            g<<"DA\n";
    }
}