Cod sursa(job #2608145)

Utilizator matthriscuMatt . matthriscu Data 30 aprilie 2020 17:51:08
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.32 kb
#include <cstdio>
#include <vector>
#include <queue>
#define INF 65536
using namespace std;

int d[50005];

struct cmp {
    bool operator()(int a, int b) {
        return d[a] > d[b];
    }
};

int main() {
    int n, m, x, y, c, nod, vecin, cost, i, inQueue[50005];
    vector<pair<int, int> > G[50005];
    priority_queue<int, vector<int>, cmp> pq;
    freopen("dijkstra.in", "r", stdin);
    freopen("dijkstra.out", "w", stdout);
    scanf("%d%d", &n, &m);

    for(i = 1; i <= m; ++i) {
        scanf("%d%d%d", &x, &y, &c);
        G[x].push_back(make_pair(y, c));
    }

    for(i = 2; i <= n; ++i) {
        d[i] = INF;
        inQueue[i] = 0;
    }
    d[1] = 0;
    pq.push(1);
    inQueue[1] = 1;

    while(!pq.empty()) {
        nod = pq.top();
        pq.pop();
        inQueue[nod] = 0;
        for(i = 0; i < G[nod].size(); ++i) {
            vecin = G[nod][i].first;
            cost = G[nod][i].second;
            if(d[vecin] > d[nod] + cost) {
                d[vecin] = d[nod] + cost;
                if(inQueue[vecin] == 0) {
                    pq.push(vecin);
                    inQueue[vecin] = 1;
                }
            }
        }
    }

    for(i = 2; i <= n; ++i)
        if(d[i] == INF)
            printf("0 ");
        else
            printf("%d ", d[i]);
}