Cod sursa(job #1261081)

Utilizator cristi23ciulica cristian cristi23 Data 11 noiembrie 2014 22:09:22
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.17 kb
#include <fstream>
#include <set>
#include <vector>
#include <limits.h>
using namespace std;
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
vector <pair<int,int> >a[50001];
multiset <pair<int,int> > b;
multiset <pair<int,int> > ::iterator it;
int x,y,z,nre[50001],n,k,i,cost,nod,d[50001];
int main()
{
    f>>n>>k;
    for(i=1;i<=n;i++){
        f>>x>>y>>z;
        a[x].push_back(make_pair(z,y));
        a[y].push_back(make_pair(z,x));
    }
    for(i=1;i<=n;i++){
        nre[i]=a[i].size();
        d[i]=INT_MAX;
    }
    b.insert(make_pair(0,1));
    d[1]=1;
    while(!b.empty()){
        it=b.begin();
        cost=(*it).first;
        nod=(*it).second;
        b.erase(it);
        for(i=0;i<nre[nod];i++){
               int nodact=a[nod][i].second;
                int costact=d[a[nod][i].second];
                if(costact>cost+a[nod][i].first){
                    d[nodact]=cost+a[nod][i].first;
                    b.insert(make_pair(d[nodact],nodact));
                }

        }
    }
    for(i=2;i<=n;i++){
        if(d[i]==INT_MAX)
            g<<0<<' ';
        else
            g<<d[i]<<' ';

    }
    return 0;
}