Cod sursa(job #3271384)

Utilizator Alex_AeleneiAlex Aelenei Ioan Alex_Aelenei Data 25 ianuarie 2025 21:40:34
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.41 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <algorithm>

using namespace std;

void dijkstra(int startNode, vector<vector<pair<int, int>>> &costMatrix, vector<bool> &visited, vector<int> &answer)
{
    queue<int> nodes;
    nodes.push(startNode);

    while (!nodes.empty())
    {
        int node = nodes.front();
        nodes.pop();

        for (auto it = costMatrix[node].begin(); it != costMatrix[node].end(); it++)
            if (!visited[it->first])
            {
                answer[it->first] = min(answer[it->first], answer[node] + it->second);
                nodes.push(it->first);
            }

        visited[node] = true;
    }
}

int main()
{
    ifstream in("dijkstra.in");
    ofstream out("dijkstra.out");

    int nodeCount, vertexCount, source, destination, cost;
    in >> nodeCount >> vertexCount;

    vector<vector<pair<int, int>>> costMatrix(nodeCount + 1, vector<pair<int, int>>());
    vector<bool> visited(nodeCount + 1, false);
    vector<int> answer(nodeCount + 1, 1e9);

    for (int vertexIndex = 1; vertexIndex <= vertexCount; vertexIndex++)
    {
        in >> source >> destination >> cost;
        costMatrix[source].push_back(pair<int, int>(destination, cost));
    }

    answer[1] = 0;
    dijkstra(1, costMatrix, visited, answer);

    for (int nodeIndex = 2; nodeIndex < answer.size(); nodeIndex++)
        out << answer[nodeIndex] << ' ';

    return 0;
}