Cod sursa(job #2375949)

Utilizator PrEaDiVviNAlexandru Hutu PrEaDiVviN Data 8 martie 2019 13:05:38
Problema Distante Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.75 kb
#include <bits/stdc++.h>

using namespace std;

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

#define nmax 50002
const int inf=(1<<31)-1;
int d[nmax];
int ver[nmax];
bool viz[nmax];

void reinitializare(int n)
{
    for(int i=1; i<=n; i++)
        viz[i]=0;
}

struct cmp
{
    bool operator()(int x,int y)
    {
        return d[x]>d[y];
    };
};

vector < pair <int, int> > v[100001];
priority_queue<int, vector<int> , cmp>q;

int n,m,t,start,p;

void dijkstra(int sursa)
{
    ///toare distante sunt initial egale cu infinit
    for(int i=1; i<=n; ++i)
        d[i]=inf;
    d[sursa]=0;
    viz[sursa]=1;
    q.push(sursa);
    while(!q.empty())
    {
        int nod=q.top();
        viz[nod]=0;
        q.pop();
        for(size_t i=0; i<v[nod].size(); ++i)
        {
            int vecin=v[nod][i].first;
            int cost=v[nod][i].second;
            if(d[vecin]>d[nod]+cost)
            {
                d[vecin]=d[nod]+cost;
                if(!viz[vecin])
                {
                    q.push(vecin);
                    viz[vecin]=1;
                }
            }
        }
    }
}

int main()
{
    f>>p;
    for(int k=1; k<=p; k++)
    {
        bool ok=1;
        f>>n>>m>>start;
        for(int i=1; i<=n; i++)
        {
            f>>ver[i];
            v[i].clear();
        }
        for(int i=1; i<=m; ++i)
        {
            int x,y,c;
            f>>x>>y>>c;
            v[x].push_back(make_pair(y,c));
        }
        dijkstra(start);
        for(int i=1; i<=n&&ok; i++)
            if(d[i]!=ver[i])ok=0;
        reinitializare(n);
        if(ok==1)g<<"DA"<<'\n';
        else g<<"NU"<<'\n';
    }
    f.close();
    g.close();
    return 0;
}