Cod sursa(job #2336337)

Utilizator denmirceaBrasoveanu Mircea denmircea Data 5 februarie 2019 01:36:05
Problema Distante Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.42 kb
#include <iostream>
#include <fstream>
#include <set>
#include <vector>
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
set < pair<int,int> >s;
vector <pair<int,int> >L[60000];
int d[60000],d2[60000];
int n,m,gr,g,i,x,y;
int nod,cost,vecin,start;
bool verif(int a[],int b[],int n){
for(int i=1;i<=n;i++)
    if(a[i]!=b[i])
    return 0;
return 1;
}
int main()
{
    int t;
    fin>>t;
       for(; t; t--)
    {
        fin>>n>>m>>start;

        for(i=1;i<=n;i++)
            fin>>d2[i];
        for(i=1; i<=m; i++)
        {
            fin>>x>>y>>gr;
            L[x].push_back({y,gr});
            L[y].push_back({x,gr});
        }
        for(i=1; i<=n; i++)
            d[i]=2000000000;
        d[start]=0;
        s.insert({0,start});
        while(s.empty()==0)
        {
            nod=s.begin()->second;
            s.erase(s.begin());
            for(i=0; i<L[nod].size(); i++)
            {
                cost=L[nod][i].second;
                vecin=L[nod][i].first;
                if(d[nod]+cost<d[vecin])
                {
                    s.erase({d[vecin],vecin});
                    d[vecin]=cost+d[nod];
                    s.insert({d[vecin],vecin});
                }
            }
        }
        if(verif(d,d2,n)==1)
            fout<<"DA\n";
        else fout<<"NU\n";
    }
        for(i=1;i<=n;i++)
        L[i].clear();
}