Cod sursa(job #2824169)

Utilizator Maria23Dutu Maria Maria23 Data 31 decembrie 2021 12:24:13
Problema Distante Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.54 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <limits>
#include <queue>

using namespace std;

const int INF = numeric_limits<int>::max();

auto cmp = [](const pair<int, int>& p1, const pair<int, int>& p2)
{
    return p1.second > p2.second;
};

vector<int> Dijkstra(int nrNoduri, const int &s, const vector<vector<pair<int,int>>> &listaAdsiCosturi) {
    vector<bool> viz(nrNoduri + 1);
    vector<int> distante(nrNoduri + 1, INF);
    priority_queue<pair<int, int>, vector<pair<int,int>>, decltype(cmp)> minHeap(cmp);  // primul element e nodul apoi distanta
    distante[s] = 0;
    minHeap.push({s, distante[s]});
    while (!minHeap.empty()){
        auto top = minHeap.top();  // (nod, distanta minima)
        minHeap.pop();
        if (viz[top.first]){
            continue;
        }
        viz[top.first] = true;
        int nodCrt = top.first;
        int distCrt = top.second;
        int nrVecini = listaAdsiCosturi[nodCrt].size();
        for (int i = 0; i < nrVecini; i++){
            int vecinCrt = listaAdsiCosturi[nodCrt][i].first;
            int costCrt = listaAdsiCosturi[nodCrt][i].second;
            if (!viz[vecinCrt] and (distCrt + costCrt < distante[vecinCrt])){
                distante[vecinCrt] = distCrt + costCrt;
                minHeap.push({vecinCrt, distante[vecinCrt]});
            }
        }
    }
    return distante;
}

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

    int nrGrafuri;
    fin >> nrGrafuri;

    for (int i = 0; i < nrGrafuri; i++){
        int nrNoduri, nrMuchii, sursa;
        fin >> nrNoduri >> nrMuchii >> sursa;
        vector<int> distanteBronzarel;

        for(int j = 0; j < nrNoduri; j++){
            int dist;
            fin >> dist;
            distanteBronzarel.push_back(dist);
        }

        vector<vector<pair<int, int>>> listaAdsiCosturi(nrNoduri + 1, vector<pair<int,int>>());
        for(int j = 0; j < nrMuchii; j++){
            int nod1, nod2, cost;
            fin >> nod1 >> nod2 >> cost;
            listaAdsiCosturi[nod1].push_back({nod2, cost});
            listaAdsiCosturi[nod2].push_back({nod1, cost});
        }

        vector<int> distanteMinime = Dijkstra(nrNoduri, sursa, listaAdsiCosturi);
        distanteMinime.erase(distanteMinime.begin());

        bool ok = true;
        for (int j = 0; j < nrNoduri; j++){
            if (distanteBronzarel[j] != distanteMinime[j]){
                ok = false;

            }
        }

        if (ok)
            fout << "DA\n";
        else
            fout << "NU\n";

    }

    return 0;
}