Cod sursa(job #2170631)

Utilizator vladbatalanBatalan Vlad vladbatalan Data 15 martie 2018 09:05:19
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.26 kb
#include <bits/stdc++.h>
#define NMAX 50010

using namespace std;

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

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

int main()
{
    int x, y, z;
    fin >> n >> m;
    for(int i=1; i<=m; i++)
    {
        fin >> x >> y >> z;
        v[x].push_back(make_pair(z, 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;
}