Cod sursa(job #1091283)

Utilizator muresan_bogdanMuresan Bogdan muresan_bogdan Data 25 ianuarie 2014 15:57:06
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.38 kb
#include<iostream>
#include<fstream>
#include<queue>
#include<vector>
using namespace std;

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

struct cuplu {
    int id, cost;
};

struct nod {
    int cost;
    bool vizitat;
    vector<cuplu> vecini;
};

bool operator<(const cuplu &a, const cuplu &b) {
    return a.cost > b.cost;
}

int n, m, i, a, b, c;
nod graf[50000];
priority_queue<cuplu> coada;

void dijkstra(int x) {
    int j;
    for(j = 0; j < graf[x].vecini.size(); j++) {
        if(!graf[graf[x].vecini[j].id].vizitat) {
            cuplu aux;
            aux.id = graf[x].vecini[j].id;
            aux.cost = graf[x].cost + graf[x].vecini[j].cost;
            coada.push(aux);
        }
    }
}

int main() {
    fin >> n >> m;
    cuplu aux;
    for(i = 0; i < m; i++) {
        fin >> a >> b >> c;
        aux.id = b;
        aux.cost = c;
        graf[a].vecini.push_back(aux);
    }
    aux.id = 1;
    aux.cost = 0;
    coada.push(aux);
    while(!coada.empty()) {
        aux = coada.top();
        coada.pop();
        if(graf[aux.id].vizitat == false) {
            graf[aux.id].cost = aux.cost;
            graf[aux.id].vizitat = true;
            dijkstra(aux.id);
        }
    }
    for(i = 2; i <= n; i++) {
        fout << graf[i].cost << '\n';
    }
    fin.close();
    fout.close();
    return 0;
}