Cod sursa(job #838263)

Utilizator sebii_cSebastian Claici sebii_c Data 19 decembrie 2012 10:46:39
Problema Algoritmul lui Dijkstra Scor 80
Compilator cpp Status done
Runda Arhiva educationala Marime 1.26 kb
#include <cstdio>
#include <cstring>

#include <algorithm>
#include <iostream>
#include <queue>
#include <vector>

using namespace std;

const int MAXN = 50100;
const int INF = (int)(1e9 + 7);

int n, m;
int d[MAXN];
bool viz[MAXN];
priority_queue<pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > > pq;
vector<int> G[MAXN], C[MAXN];

void dijkstra()
{
    memset(viz, false, sizeof(viz));
    for (int i = 2; i <= n; ++i) 
        d[i] = INF;
    pq.push(make_pair(0, 1));
    while (!pq.empty()) {
        int x = pq.top().second;
        int val = pq.top().first;
        pq.pop();
        if (viz[x]) continue;
        viz[x] = true;
        for (size_t i = 0; i < G[x].size(); ++i)
            if (C[x][i] + val < d[G[x][i]]) {
                d[G[x][i]] = val + C[x][i];
                pq.push(make_pair(d[G[x][i]], G[x][i]));
            }
    }
}

int main()
{
    freopen("dijkstra.in", "r", stdin);
    freopen("dijkstra.out", "w", stdout);

    cin >> n >> m;
    for (int i = 1; i <= m; ++i) {
        int a, b, c;
        cin >> a >> b >> c;
        G[a].push_back(b);
        C[a].push_back(c);
    }
    dijkstra();

    for (int i = 2; i <= n; ++i)
        if (d[i] >= INF) cout << 0 << " ";
        else cout << d[i] << " ";
    cout << "\n";

    return 0;
}