Cod sursa(job #3215775)

Utilizator victor_gabrielVictor Tene victor_gabriel Data 15 martie 2024 12:44:36
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.16 kb
#include <fstream>
#include <vector>
#include <queue>

using namespace std;

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

typedef pair<int, int> Pair;
const int DIM = 50010;

int n, m, x, y, c;
int dist[DIM];
vector<pair<int, int>> l[DIM];
priority_queue<Pair, vector<Pair>, greater<Pair>> pq;
bool visited[DIM];

void dijkstra(int start) {
    dist[start] = 0;
    pq.emplace(dist[start], start);
    while (!pq.empty()) {
        int node = pq.top().second;
        pq.pop();
        if (visited[node])
            continue;

        visited[node] = true;
        for (auto adjElem : l[node]) {
            int adjNode = adjElem.first;
            int adjDist = adjElem.second;
            if (dist[adjNode] > dist[node] + adjDist) {
                dist[adjNode] = dist[node] + adjDist;
                pq.emplace(dist[adjNode], adjNode);
            }
        }
    }
}

int main() {
    fin >> n >> m;
    for (int i = 1; i <= m; i++) {
        fin >> x >> y >> c;
        l[x].emplace_back(y, c);
    }

    for (int i = 1; i <= n; i++)
        dist[i] = INT_MAX;
    dijkstra(1);

    for (int i = 2; i <= n; i++)
        fout << dist[i] << ' ';

    return 0;
}