Cod sursa(job #2139480)

Utilizator andrei.gramescuAndrei Gramescu andrei.gramescu Data 22 februarie 2018 16:21:29
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.33 kb
#include <cstdio>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
#define NMAX 50005
#define INFINIT 50005

typedef pair<int,int>ii;
vector<ii> a[NMAX];
int n, m, sol[NMAX], point[NMAX];
int x, y, i, d;
ii temp;
 
void dj(){
    int d, nod, i, child, cost;
    priority_queue<ii, vector<ii>, greater<ii> >q;
    q.push(ii(0, 1));
 
    while(!q.empty()){
        d = q.top().first;
        nod  = q.top().second;
        q.pop();
 
        if(sol[nod] < d)
            continue;
 
        for(i=0; i<a[nod].size(); i++){
            ///a[nod][j].first = nodul
            ///a[nod][j].second = costul
 
            child = a[nod][i].first;
            cost = a[nod][i].second;
 
            if(sol[child] > d + cost){
                sol[child] = d + cost;
                q.push(ii(sol[child], child));
            }
        }
    }
}
 
int main(){
 
    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, &d);
			
        a[x].push_back(ii(y, d));
    }
 
    for(int x=1; x<=n; x++)
			sol[x] = INFINIT;
    dj();
	
    for(i=2; i<=n; i++){
				if(sol[i] == INFINIT)
					printf("0 ");
				else
        printf("%d ", sol[i]);
    }
    return 0;
}