Cod sursa(job #2425809)

Utilizator ZamfiAndreiZamfira Andrei ZamfiAndrei Data 25 mai 2019 03:50:10
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.26 kb
#include <fstream>
#include <list>
#include <queue>

using namespace std;

#define INF (1<<25)
typedef pair<int, int> iPair;

void Dijkstra(int source, int vertices, list<iPair> *adj) {
    ofstream g("dijkstra.out");
    priority_queue<iPair, vector<iPair>, greater<iPair>> priorityQueue;
    vector<int> dist (vertices, INF);

    priorityQueue.push(make_pair(0, source));
    dist[source] = 0;

    while (!priorityQueue.empty()) {
        int u = priorityQueue.top().second;
        priorityQueue.pop();

        list<iPair>::iterator i;
        for (i = adj[u].begin(); i != adj[u].end(); ++i) {
            int v = (*i).first;
            int weight = (*i).second;

            if (dist[v] > dist[u] + weight) {
                dist[v] = dist[u] + weight;
                priorityQueue.push(make_pair(dist[v], v));
            }
        }
    }

    for (int i = 1; i < vertices; i++) {
        g << dist[i] << " ";
    }
}

int main() {
    ifstream f("dijkstra.in");
    int vertices, edges, from, to, weight;
    f >> vertices >> edges;
    list<iPair> *adj = new list<iPair> [vertices];

    for (int i = 0; i < edges; i++) {
        f>>from;
        f>>to;
        f>>weight;
        adj[from - 1].push_back(make_pair(to - 1, weight));
    }

    Dijkstra(0, vertices, adj);
}