Cod sursa(job #1458794)

Utilizator glbglGeorgiana bgl glbgl Data 8 iulie 2015 14:28:44
Problema Algoritmul lui Dijkstra Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.99 kb
#include <stdio.h>
#include <fstream>
#include <vector>
#include <algorithm>
#include <iterator>
 
#define INF 1<<30
 
using namespace std;
 
ifstream in("dijkstra.in");
ofstream out("dijkstra.out");
 
 
int N, M;
vector< vector< pair<int,int> > > neigh;
vector<int> minHeap;
vector<int> d;
vector<int> visited;
 
 
 
struct comparator{
    bool operator()(int x, int y){
        return d[x] > d[y];
         
    }
};
 
void read(){
 
    in >> N >> M;
 
    d.resize(N+1);
    d.assign(N+1, INF);
 
    visited.resize(N+1);
 
    for(int i = 0; i <= N; ++i){
        vector< pair<int,int> > v;
        neigh.push_back(v);
    }
 
    int x, y, c;
 
    for(int i = 0; i < M; ++i){
        in >> x >> y >> c;
        neigh[x].push_back(make_pair(y,c));
        
    }
 
}
 
 
void Dijkstra(){

    make_heap(minHeap.begin(), minHeap.end(), comparator());

    for(unsigned int i = 0; i < neigh[1].size(); ++i){
        minHeap.push_back(neigh[1][i].first);
        push_heap(minHeap.begin(), minHeap.end(), comparator());
        d[neigh[1][i].first] = neigh[1][i].second;
        visited[neigh[1][i].first] = 1;
    }
 
    visited[1] = 2;
 
    while(!minHeap.empty()){
 
        int u = minHeap.front();
 
        pop_heap(minHeap.begin(), minHeap.end(), comparator());
        minHeap.pop_back();
 
        for(int i = 0; i < neigh[u].size(); ++i){
            int v = neigh[u][i].first;
            int dist = d[u] + neigh[u][i].second;
 
            if(d[v] > dist){
                d[v] = dist;
                if(visited[v]){
                    push_heap(minHeap.begin(), minHeap.end(), comparator());
                }
                else{
                    minHeap.push_back(v);
                    push_heap(minHeap.begin(), minHeap.end(), comparator());
                    visited[v] = 1;
                }
            }
        }
    }
}
 
 
int main(){
 
    read();
    Dijkstra();
 
    for(int i = 2; i <=N; ++i){
        if(d[i] == INF)
            out << 0 << " ";
        else out << d[i] << " ";
    }
 
}