Cod sursa(job #2037301)

Utilizator DawlauAndrei Blahovici Dawlau Data 11 octombrie 2017 23:14:16
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.21 kb
#include<fstream>
#include<set>
#include<vector>
#include<cstring>
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
const int NMAX=50005,inf=0x3f3f3f3f;
vector<pair<int,int>>v[NMAX];
int drum[NMAX];
int n,m;
void citire(){
    int i,x,y,c;
    fin>>n>>m;
    for(i=1;i<=m;++i){
        fin>>x>>y>>c;
        v[x].push_back(make_pair(y,c));
    }
}
void dijkstra(){
    int i;
    memset(drum,inf,sizeof drum);
    drum[1]=0;
    set<pair<int,int>>s;
    s.insert(make_pair(0,1));
    while(!s.empty()){
        int nod=s.begin()->second;
        s.erase(s.begin());
        for(int j=0;j<v[nod].size();++j){
            int cost=v[nod][j].second;
            int vecin=v[nod][j].first;
            if(drum[nod]+cost<drum[vecin]){
                if(drum[vecin]!=inf)
                    s.erase(s.find(make_pair(drum[vecin],vecin)));
                drum[vecin]=drum[nod]+cost;
                s.insert(make_pair(drum[vecin],vecin));
            }
        }
    }
}
void afisare(){
    int i;
    for(i=2;i<=n;++i){
        if(drum[i]==inf)
            drum[i]=0;
        fout<<drum[i]<<' ';
    }
}
int main(){
    citire();
    dijkstra();
    afisare();
}