Cod sursa(job #1112657)

Utilizator petiVass Peter peti Data 19 februarie 2014 22:14:57
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.01 kb
#include <iostream>
#include <vector>
#include <fstream>
#include <queue>
#include <list>
using namespace std;

struct node{int b,c;};

int main(){
    ifstream ifs("dijkstra.in");
    ofstream ofs("dijkstra.out");
    int N,M;
    ifs>>N>>M;

    vector<list<node> > g(N);

    for(int i=0;i<M;i++){
        int a;
        node t;
        ifs>>a>>t.b>>t.c;t.b--;a--;
        g[a].push_back(t);
    }

    queue<int> q; q.push(0);
    vector<int> m(N,-1);m[0]=0;
    vector<bool> v(N,false);
    while(!q.empty()){
        int a=q.front();
        q.pop();
        v[a]=true;
        for(list<node>::iterator it=g[a].begin();it!=g[a].end();it++){
            if((m[a]+it->c<m[it->b])||(m[it->b]==-1)){
                m[it->b]=m[a]+it->c;
                if(!v[it->b]){
                    q.push(it->b);
                    v[it->b]=true;
                }
            }
        }
    }

    for(int i=1;i<m.size();i++)
        if(-1==m[i])ofs<<0<<" ";
        else ofs<<m[i]<<" ";
}