Cod sursa(job #3242678)

Utilizator raizoSoare Antonio raizo Data 13 septembrie 2024 13:00:12
Problema Algoritmul Bellman-Ford Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.72 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;

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

int n, m, x, y, cost, start;

vector<int> bellman_ford(vector<vector<pair<int, int>>>& list, int start) {

    queue<int> q;
    vector<int> dist(n + 1, -1);
    vector<int> parent(n + 1, 0);
    vector<int> in_queue(n + 1, 0);
    vector<int> nr_of_visits(n + 1, 0);

    dist[start] = 0;
    q.push(start);
    in_queue[start] = 1;
    nr_of_visits[start] = 1;

    while (!q.empty()) {
        int node = q.front();
        q.pop();
        in_queue[node] = 0;
        
        for (auto& e : list[node]) {
            if (dist[e.first] == -1 || dist[node] + e.second < dist[e.first]) {
                dist[e.first] = dist[node] + e.second;
                parent[e.first] = node;
                
                if (in_queue[node] == 0) {
                    in_queue[node] = 1;
                    q.push(e.first);
                    nr_of_visits[node]++;
                    if (nr_of_visits[node] == list.size()) { //cycle
                        dist[0] = -2;
                        return dist;
                    }
                }
            }
        }
    }
    return dist;
}


int main()
{
    int start = 1;
    cin >> n >> m;
    vector<vector<pair<int, int>>> list(n + 1);

    for (int i = 0; i < m; i++) {
        cin >> x >> y >> cost;
        list[x].push_back({ y,cost });
    }

    vector<int> dist = bellman_ford(list, start);

    if (dist[0] == -2) {
        cout << "Ciclu negativ!" << endl;
    }
    else {
        for (int i = 2; i < dist.size(); i++) {
            cout << dist[i] << " ";
        }
    }

}