Cod sursa(job #2963954)

Utilizator PalcPalcu Stefan Palc Data 12 ianuarie 2023 10:41:22
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.09 kb
#include <bits/stdc++.h>
using namespace std;

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

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

void Dijk(int x)
{
    d[x] = 0;
    q.push({0, x});
    while (!q.empty())
    {
        x = q.top().second;
        q.pop();
        if (viz[x] == 0)
            for(auto e : a[x])
                if (d[e.second] > d[x] + e.first)
                {
                    d[e.second] = d[x] + e.first;
                    if (viz[e.second] == 0) q.push({-d[e.second], e.second});
                }
        viz[x] = 1;
    }
}

int main()
{
    int x, y, c;
    fin >> n >> m;
    for (int i = 1; i <= m; i++)
    {
        fin >> x >> y >> c;
        a[x].push_back({c, y});
    }
    for (int i = 1; i <= n; i++)
        d[i] = 1e9;
    Dijk(1);
    for (int i = 2; i <= n; i++)
    {
        if (d[i] != 1e9)
            fout << d[i] << " ";
        else fout << "0 ";
    }
    fout << "\n";
    fin.close();
    fout.close();
    return 0;
}