Cod sursa(job #3268487)

Utilizator Andrei24543Andrei Hulubei Andrei24543 Data 15 ianuarie 2025 14:48:34
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include <bits/stdc++.h>
#define oo 1000000005
using namespace std;

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

int n, m, d[50005];
vector<pair<int, int>> l[50005];
bitset<50005> viz;
priority_queue<pair<int, int>> q;

void Dijkstra(int start)
{
    int k, cost;
    q.push({0, start});
    d[start] = 0;
    for (int i = 2; i <= n; i++)
        d[i] = oo;
    while (!q.empty())
    {
        k = q.top().second;
        q.pop();
        if (viz[k] == 0)
        {
            viz[k] = 1;
            for (auto e : l[k])
                if (d[e.first] > d[k] + e.second)
                {
                    d[e.first] = d[k] + e.second;
                    q.push({-d[e.first], e.first});
                }
        }
    }
}

int main()
{
    int i, j, cost;
    fin >> n >> m;
    while (m--)
    {
        fin >> i >> j >> cost;
        l[i].push_back({j, cost});
    }

    Dijkstra(1);

    for (i = 2; i <= n; i++)
        if (d[i] == oo) fout << "0 ";
        else fout << d[i] << " ";
    return 0;
}