Cod sursa(job #2572521)

Utilizator PopeangaMihneaPopeanga Mihnea- Stefan PopeangaMihnea Data 5 martie 2020 13:11:18
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.17 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];
const int INF = 1e9;
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] = INF;
    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)
    {
        if(d[i] != INF) fout << d[i] << " ";
        else fout << "0 ";
    }
    fout << "\n";
    return 0;
}