Cod sursa(job #2076933)

Utilizator sebi_andrei2008Lazar Eusebiu sebi_andrei2008 Data 27 noiembrie 2017 14:06:44
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.72 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <utility>
#include <set>
#define INFINITY 2147483647
#define MAX_N 50001
using namespace std;

ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
vector<pair<int, int>> graf[MAX_N];
int dist[MAX_N];
int n, m;
int source;
set<pair<int, int>> q;

int main() {
    fin>>n>>m;
    source = 1;
    
    for (int i = 0; i < m; i++) {
        int a, b, c;
        fin>>a>>b>>c;
        graf[a].push_back(make_pair(c, b));
    }
    
    for (int i = 1; i <= n; i++) {
        dist[i] = INFINITY;
        if (i == source)
            dist[i] = 0;
        q.insert(make_pair(dist[i], i));
    }
    
    while (!q.empty()) {
        pair<int, int> nod = *q.begin();
        q.erase(q.begin());
        
        vector<pair<int, int>> neighbours = graf[nod.second];
        vector<pair<int, int>>::iterator it;
        for (it = neighbours.begin(); it != neighbours.end(); it++) {
            pair<int, int> neighbour = *it;
            if (dist[nod.second] != INFINITY) {
                if (q.find(make_pair(dist[neighbour.second], neighbour.second)) != q.end()) {
                    int alt = nod.first + neighbour.first;
                    if (alt < dist[neighbour.second]) {
                        q.erase(make_pair(dist[neighbour.second], neighbour.second));
                        dist[neighbour.second] = alt;
                        q.insert(make_pair(dist[neighbour.second], neighbour.second));
                    }
                }
            }
        }
    }
    
    for (int i = 1; i <= n; i++) {
        if (i != source) {
            if (dist[i] == INFINITY)
                fout<<"0 ";
            else
                fout<<dist[i]<<" ";
        }
    }
    return 0;
}