Cod sursa(job #2441366)

Utilizator 1000Sabin Ilegitim 1000 Data 20 iulie 2019 13:14:48
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.45 kb
#include <fstream>
#include <utility>
#include <queue>
#include <vector>
#include <bitset>

using namespace std;

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

vector <vector <pair <int, int>>> a;
priority_queue <pair <int, long long>> que;
bitset <100005> v;

long long d[100005];

int n, m, p;

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

    p = 1;

    a.resize(n + 2);

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

        cin >> x >> y >> c;

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

    for(int i = 1; i <= n; i++)
        d[i] = 1e16;

    d[p] = 0;

    que.push(make_pair(0, p));

    while(!que.empty())
    {
        int x = que.top().second;
        que.pop();

        while(!que.empty() and v[x] == 1)
        {
            x = que.top().second;
            que.pop();
        }

        v[x] = 1;

        for(int i = 0; i < a[x].size(); i++)
        {
            if(1LL * a[x][i].second + d[x] < d[a[x][i].first])
            {
                d[a[x][i].first] = 1LL * a[x][i].second + d[x];
                que.push(make_pair(-d[a[x][i].first], a[x][i].first));
            }
        }
    }

    for(int i = 1; i <= n; i++)
        if(i != p)
        {
            if(d[i] != 1e16)
                cout << d[i] << " ";
            else
                cout << "0 ";
        }

    cout << '\n';

    return 0;
}