Cod sursa(job #1906842)

Utilizator savigunFeleaga Dragos-George savigun Data 6 martie 2017 16:39:36
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.28 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <set>
#include <tuple>
using namespace std;

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


int n, m, d[50001];
bool viz[50001];
vector< pair<int, int> > g[50001];
set< pair<int, int> > q;
const int INF = 2e9 + 1;

void citire();


int main()
{
    citire();
    q.insert({0, 1});

    while (!q.empty()) {
        int x = q.begin() -> second;
        int cost = q.begin() -> first;
        q.erase(q.begin());
        if (viz[x]) continue;

        viz[x] = true;
        for (int i = 0; i < g[x].size(); ++i) {
            int y = g[x][i].second;
            if (!viz[y]) {
                if (cost + g[x][i].first < d[y]) {
                    d[y] = cost + g[x][i].first;
                    q.insert({d[y], y});
                }
            }
        }
    }

    for (int i = 2; i <= n; ++i) {
        if (d[i] == INF) {
            out << 0 << " ";
        } else {
            out << d[i] << " ";
        }
    }

    return 0;
}


void citire() {
    in >> n >> m;
    for (int i = 1, x, y, cost; i <= m; ++i) {
        in >> x >> y >> cost;
        g[x].push_back({cost, y});
    }

    for (int i = 1; i <= n; ++i)
        d[i] = INF;

    in.close();
}