Cod sursa(job #560974)

Utilizator mihaimoldovanMihai Moldovan mihaimoldovan Data 18 martie 2011 19:36:18
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 kb
#include<vector>
#include<fstream>
using namespace std;

#define nn 50001
#define pb push_back
#define mp make_pair
#define infinit 1<<10
vector<int,int>g[nn];
int n,m,d[nn];

void citire(){
    int a,b,c;
    ifstream fin("dijsktra.in");
    fin>>n>>m;
    for(;m;--m){
        fin>>a>>b>>c;
        g[a].pb(mp(b,c));
    }
}
void afisare(){
    FILE *f=fopen("dijkstra.out");
    for(int i=2;i<=n;++i)
        fprintf(f,"%d ",d[i])
}
void bellmann(int sursa){
    for(int i=1;i<=n;++i)
        d[i]=infinit;
    d[sursa]=0;
    int coada[nn],st,dr,i;
    st=dr=1;
    coada[1]=sursa;
    while(st<=dr){
        int k=coada[st++];
        for(int i=1;i<g[k].size();++i){
            int kk=g[k].first();
            int cost=g[k].second();
            if(d[k]+cost<d[kk]){
                d[kk]=d[k]+cost;
                coada[++dr]=kk;
            }
        }
    }
}

int main(){
    citire();
    bellmann(1);
    afisare();
    return 0;
}