Cod sursa(job #1597930)

Utilizator grimmerFlorescu Luca grimmer Data 12 februarie 2016 14:52:16
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.16 kb
#include <cstdio>
#include <vector>
#include <queue>
using namespace std;

const int NMAX = 50005;
const int INF = 250000000;

typedef pair<int, int> ii;
typedef vector<int> vi;
typedef vector<ii> vii;

vector<ii> G[NMAX+5];

int main()
{
    int i, n, m, u, v, w, s;
    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",&u,&v,&w);
        G[u].push_back(ii(v,w));
        G[v].push_back(ii(u,w));
    }
    s = 1;
    vi dist(n+1, INF);
    dist[s] = 0;
    priority_queue< ii, vector<ii>, greater<ii> > pq;
    pq.push(ii(0,s));

    while(!pq.empty()){

        ii front = pq.top();
        pq.pop();
        int d = front.first, u = front.second;
        if(d > dist[u])
            continue;
        for(int j = 0; j < (int)G[u].size(); ++j){
            int v = G[u][j].first, duv = G[u][j].second;
            if(dist[u] + duv < dist[v]){
                dist[v] = dist[u] + duv;
                pq.push(ii(dist[v], v));
            }
        }
    }
    for(int i = 2; i <= n; ++i)
        printf("%d ", dist[i]);
    return 0;
}