Cod sursa(job #2173644)

Utilizator vladbatalanBatalan Vlad vladbatalan Data 15 martie 2018 23:18:07
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.25 kb
#include <bits/stdc++.h>

using namespace std;

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

const int oo = 100000000;
int n, m, d[50010];
bitset<50010> in_set;
vector<pair<int, int> > v[50010]; /// cost, pana la nodul
set<pair<int, int> > s;
pair<int, int> top;

int main()
{
    int x, y, cost, nod;
    fin >> n >> m;
    while(m--)
    {
        fin >> x >> y >> cost;
        v[x].push_back(make_pair(cost, y));
    }
    for(int i=1; i<=n; i++) d[i] = oo;
    d[1] = 0;
    s.insert(make_pair(0, 1));
    in_set[1] = 1;
    while(!s.empty())
    {
        top = *s.begin();
        s.erase(s.begin());
        in_set[top.second] = 0;
        for(auto vec:v[top.second])
        {
            if(d[vec.second] > d[top.second] + vec.first)
            {
                if(in_set[vec.second])
                    s.erase(s.find(make_pair(d[vec.second], vec.second)));
                d[vec.second] = d[top.second] + vec.first;
                s.insert(make_pair(d[vec.second], vec.second));
                in_set[vec.second] = 1;
            }
        }
    }
    for(int i=2; i<=n; i++)
    {
        if(d[i] == oo)
            fout<<0<<' ';
        else
            fout<<d[i]<<' ';
    }
    return 0;
}