Cod sursa(job #2572507)

Utilizator PopeangaMihneaPopeanga Mihnea- Stefan PopeangaMihnea Data 5 martie 2020 13:08:42
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.08 kb
#include <bits/stdc++.h>

using namespace std;

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

int n, m;
int d[50001];
bool viz[50001];
vector<pair<int, int>>G[50001];
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>>PQ;

int main()
{
    fin >> n >> m;

    for(int i = 1; i <= m; ++i)
    {
        int x, y, c;
        fin >> x >> y >> c;
        G[x].push_back(make_pair(y, c));
    }

    for(int i = 1; i <= n; ++i) d[i] = INT_MAX;
    d[1] = 0;
    PQ.push({0, 1});

    while(!PQ.empty())
    {
        pair<int, int> nod = PQ.top();
        PQ.pop();
        if(!viz[nod.second])
        {
            for(auto it: G[nod.second])
            {
                if(d[it.first] > it.second + d[nod.second])
                {
                    d[it.first] = it.second + d[nod.second];
                    PQ.push({d[it.first], it.first});
                }
            }
            viz[nod.second] = 1;
        }
    }

    for(int i = 2; i <= n; ++i) fout << d[i] << " ";
    fout << "\n";
    return 0;
}