Cod sursa(job #1976930)

Utilizator blatulInstitutul de Arta Gastronomica blatul Data 4 mai 2017 16:26:03
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.08 kb
#include <bits/stdc++.h>
using namespace std;

#define NMAX 50002

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

vector< pair<int, int> > graf[NMAX];

int dist[NMAX];

void Dijkstra(const int N, const int start) {

    priority_queue< pair<int, int> > Heap;

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

    dist[start] = 0;

    Heap.push(make_pair(-dist[start], start));

    while (!Heap.empty()) {

        int nod = Heap.top().second;
        Heap.pop();

        for (auto adj: graf[nod]) {

            if (dist[adj.first] > dist[nod] + adj.second) {

                dist[adj.first] = dist[nod] + adj.second;
                Heap.push(make_pair(-dist[adj.first], adj.first));
            }
        }
    }
}

int main() {

    int N, M;

    fin >> N >> M;

    int x, y, c;
    while (M--) {

        fin >> x >> y >> c;

        graf[x].push_back(make_pair(y, c));
    }

    Dijkstra(N, 1);

    for (int i = 2; i <= N; ++i)
        fout << (dist[i] == INT_MAX ? 0 : dist[i]) << " ";

    return 0;
}