Cod sursa(job #3226705)

Utilizator _andrei4567Stan Andrei _andrei4567 Data 22 aprilie 2024 17:07:44
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.22 kb
#include <fstream>
#include <queue>
#include <vector>

using namespace std;

ifstream cin ("dijkstra.in");
ofstream cout ("dijkstra.out");

const int INF = 1e9;

const int N = 5e4;

int dist[N + 1];

int m, n, x, y, cost;

struct ceva
{
    int node, cost;
    bool operator < (const ceva &a)const
    {
        return a.cost < cost;
    }
};

priority_queue <ceva> q;
vector <ceva> g[N + 1];

void dijkstra (int node)
{
    for (int i = 1; i <= n; ++i)
        dist[i] = INF;
    dist[node] = 0;
    q.push({node, 0});
    while (!q.empty())
    {
        ceva x = q.top();
        q.pop();
        if (dist[x.node] < x.cost)continue;
        for (auto it : g[x.node])
            if (dist[it.node] > dist[x.node] + it.cost)
            dist[it.node] = dist[x.node] + it.cost, q.push({it.node, dist[it.node]});
    }
    for (int i = 2; i <= n; ++i)
        if (dist[i] == INF)
            cout << "0 ";
        else
            cout << dist[i] << ' ';
}

int main()
{
    ios_base :: sync_with_stdio(false);
    cin.tie(0);
    cin >> n >> m;
    for (int i = 1; i <= n; ++i)
    {
        cin >> x >> y >> cost;
        g[x].push_back({y, cost});
    }
    dijkstra (1);
    return 0;
}