Cod sursa(job #1605499)

Utilizator PetrutiuPaulPetrutiu Paul Gabriel PetrutiuPaul Data 19 februarie 2016 08:29:17
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.09 kb
#include <iostream>
#include <fstream>
#include <set>
#include <vector>

    using namespace std;

    ifstream fin("dijkstra.in");
    ofstream fout("dijkstra.out");

vector<int> v[50002];
vector<int> c[50002];
set < pair< int, int > > s;
int g[50002];
int n,m,dist;

void dijkstra(int nod)
{
    s.insert(make_pair(0,nod));
    g[nod]=0;

    while(!s.empty())
    {
        nod=(*s.begin()).second;
        dist=(*s.begin()).first;

        for(int i=0;i<v[nod].size();i++)
        {
            if(g[v[nod][i]]>dist+c[nod][i])
            {
                g[v[nod][i]]=dist+c[nod][i];
                s.insert(make_pair(g[v[nod][i]],v[nod][i]));
            }
        }
        s.erase(*s.begin());
    }
}

int main()
{
    int x,y,cost;

    fin>>n>>m;

    for(int i =1;i<=n;i++)
    {
        fin>>x>>y>>cost;
        v[x].push_back(y);
        c[x].push_back(cost);
    }


    for(int i=1; i<=n;i++)
        g[i]=(1<<30)-1;

    dijkstra(1);

    for(int i=2;i<=n;i++)
        if(g[i]==(1<<30)-1)fout<<-1<<' ';
        else fout<<g[i]<<' ';


}