Cod sursa(job #1911946)

Utilizator MailatMailat Radu Mailat Data 7 martie 2017 22:21:34
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.42 kb
#include <bits/stdtr1c++.h>
#define INF 500010
#define maxn 500001
using namespace std;

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

vector <int> A[maxn];
vector <int> C[maxn];
int d[maxn], V[maxn];
int N, M, x0 = 1;

void addEdge(int x, int y, int c) {
    A[x].push_back(y);
    C[x].push_back(c);
    A[y].push_back(x);
    C[y].push_back(c);
}

void init() {
    int x, y, c;
    fin >> N >> M;
    for(int i=0; i < M; i++) {
        fin >> x >> y >> c;
        addEdge(x, y, c);
    }
    for(int i = 0; i <=N; i++) d[i] = INF;
    d[x0] = 0;
    for(int j = 0; j < A[x0].size(); j++) {
        d[A[x0][j]] = C[x0][j];
    }
}

void dijkstra() {
    int dMin, VfMin;
    for(int j=1; j<N; j++) {
        dMin = INF;
        for(int i=1; i<=N; i++) {
            if(!V[i] && dMin > d[i]) {
                dMin = d[i];
                VfMin = i;
            }
        }
        V[VfMin] = 1;
        for(int i=1; i <=N; i++) {
            if(!V[i]){
                for(int k=0; k < A[VfMin].size(); k++) {
                    if(A[VfMin][k] == i) {
                        if(d[i] > dMin + C[VfMin][k]) {
                            d[i] = dMin + C[VfMin][k];
                        }
                    }
                }
            }
        }

    }
}

int main()
{
    init();
    dijkstra();
    for(int i=2; i <= N; i++ ) fout << d[i] << " ";
    return 0;
}