Dijsktra Al go с использованием набора в C ++ - PullRequest
Новая
       29

Dijsktra Al go с использованием набора в C ++

0 голосов
/ 18 апреля 2020

Вот код для реализации алгоритма Дейкстры с использованием Set в C ++. В алгоритме dijsktra мы должны найти вершину с минимальным весом на каждой итерации и попытаться ослабить ее соседей. В этом коде каждый сосед, который расслабляется после вставки источника в набор. Вершина просто выбирается с помощью s.begin (). Как это гарантирует, что у нас есть вершина с минимумом d [v]? 

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef pair<int, int> pi;
typedef pair<ll, ll> pl;
typedef vector<int> vi;
typedef vector<ll> vl;
typedef vector<double> vd;
typedef vector<bool> vb;
typedef vector<char> vc;
typedef vector<string> vs;
typedef vector<pi> vp;
typedef vector<pl> vpl;

    const ll inf = 10000000000000ll;

    vector<pair<int,int> > e[100000];
    vector<int> put;
    set<pair<ll,int> > s;

    ll d[100000];
    int p[100000];

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

        for(int i = 0; i < m; ++i) {
            int v1, v2, w;
            cin >> v1 >> v2 >> w;
            v1--;
            v2--;
            e[v1].push_back(make_pair(v2,w));
            e[v2].push_back(make_pair(v1,w));
        }

        for(int i = 0; i < n; ++i)
            d[i] = inf;
        d[0] = 0;

    s.insert(make_pair(d[0], 0));
    while (!s.empty()) {
        int v = s.begin()->second;
        s.erase(s.begin());

        for(int i = 0; i < (int(e[v].size())); ++i) {
            int prvi = e[v][i].first;
            int w = e[v][i].second;

            if (d[v] + w < d[prvi]) {
                s.erase(make_pair(d[prvi], prvi));
                d[prvi] = d[v]+  w;
                p[prvi] = v;
                s.insert(make_pair(d[prvi], prvi));
            }
        }
    }
    if (d[n - 1] == inf)
        cout << -1;
    else {
        int c = n - 1;
        while (c != 0) {
            put.push_back(c + 1);
            c = p[c];
        }

        put.push_back(1);
        for(int i = (put.size()) - 1; i >= 0; --i)
        cout << put[i] << ' ';
    }


}
...