Cod sursa(job #2476881)

Utilizator uvIanisUrsu Ianis Vlad uvIanis Data 19 octombrie 2019 12:06:49
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.42 kb
#include <bits/stdc++.h>
#define NMAX 50001
#define INFNITY 1800000000000
using namespace std;

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

int n, m;

vector<vector<pair<int, int>>> graph(NMAX, vector<pair<int, int>>());

vector<bool> visited(NMAX, false);
vector<long long> dist(NMAX, INFNITY);
vector<bool> inQueue(NMAX, false);

struct comp{
    bool operator ()(int& A, int &B)
    {
        if(dist[A] > dist[B]) return true;
        return false;
    }
};

priority_queue<int, vector<int>, comp> pq;


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

    for(int x, y, c, i = 1; i <= m; ++i)
    {
        fin >> x >> y >> c;

        graph[x].push_back(make_pair(y, c));
    }

    dist[1] = 0;
    pq.push(1);

    while(!pq.empty())
    {
        int node = pq.top();
        pq.pop();

        inQueue[node] = false;

        visited[node] = true;

        for(auto& next : graph[node])
        {
            if(dist[node] + next.second < dist[next.first])
            {
                dist[next.first] = dist[node] + next.second;
                if(inQueue[next.first] == false)
                {
                     pq.push(next.first);
                     inQueue[next.first] = true;
                }

            }
        }
    }

    for(int i = 2; i <= n; ++i)
    {
        if(dist[i] == INFNITY) fout << "0 ";
        else fout << dist[i] << " ";
    }
}