Cod sursa(job #2448413)

Utilizator Stefan_PiscuPiscu Stefan Constantin Stefan_Piscu Data 16 august 2019 20:34:57
Problema Distante Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.63 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;

ifstream fin("distante.in");
ofstream fout("distante.out");

const int MAX=50007;
int t,n,m,s,x,y,c,dist[MAX],d[MAX],a,b,cost;
bool sel[MAX];
const int INF=999999999;

vector<pair<int ,int > >graf[MAX];
priority_queue<pair<int, int > >h;

void dijkstra(int start)
{
    for(int i=1;i<=n;i++)
    {
        d[i]=INF;
        sel[i]=false;

    }
    d[start]=0;
    h.push(make_pair(-d[start],start));
    while(!h.empty())
    {
        while(!h.empty() && sel[h.top().second]) h.pop();

        if(h.empty()) return;

        x=h.top().second;
        h.pop();
        sel[x]=true;

        for(int i=0;i<graf[x].size();i++)
        {
            pair<int , int >p=graf[x][i];
            y=p.second;
            c=p.first;

            if(d[x]+c<d[y])
            {
                d[y]=d[x]+c;
                h.push(make_pair(-d[y],y));

                            }
        }
    }
}
int main()
{
    fin>>t;
    for(int i=1;i<=t;i++)
    {
        fin>>n>>m>>s;
        for(int j=1;j<=n;j++) fin>>dist[j];
        for(int j=1;j<=m;j++)
        {
            fin>>a>>b>>cost;
            graf[a].push_back(make_pair(cost,b));
            graf[b].push_back(make_pair(cost,a));

        }
        dijkstra(s);
        int ok=0;
        for(int j=1;j<=n;j++)
        {
            if(dist[j]!=d[j])
            {
                ok=1;
                break;
            }
        }
        if(ok==0) fout<<"DA\n";
        else fout<<"NU\n";
        for(int j=1;j<=n;j++) graf[j].clear();

    }
    return 0;
}