Cod sursa(job #1732401)

Utilizator SolcanMihaiSolcan Mihai Andrei SolcanMihai Data 21 iulie 2016 16:16:45
Problema Algoritmul Bellman-Ford Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.49 kb
#include <cstdio>
#include <vector>
#include <queue>

using namespace std;

int n, m;
vector<pair<int, int> > vecini[50500];
queue<int> Q;
int viz[50500];
int dist[50500];

const int inf = 1<<20;

void citire()
{
    int tmp1, tmp2, tmp3;

    scanf("%d %d", &n, &m);

    for(int i = 0; i < m; i++)
    {
        scanf("%d %d %d", &tmp1, &tmp2, &tmp3);
        tmp1--;
        tmp2--;

        vecini[tmp1].push_back(make_pair(tmp3, tmp2));
    }

    for(int i = 1; i < n; i++)
    {
        dist[i] = inf;
    }
}

bool bellman()
{
    Q.push(0);

    while(!Q.empty())
    {
        int y = Q.front();
        Q.pop();

        int l = vecini[y].size();

        for(int i = 0; i < l; i++)
        {
            if(dist[vecini[y][i].second] > dist[y] + vecini[y][i].first)
            {
                dist[vecini[y][i].second] = dist[y] + vecini[y][i].first;
                viz[vecini[y][i].second]++;

                if(viz[vecini[y][i].second] > n)
                {
                    return false;
                }

                Q.push(vecini[y][i].second);
            }
        }
    }

    return true;
}

int main()
{
    freopen("bellmanford.in", "r", stdin);
    freopen("bellmanford.out", "w", stdout);

    citire();
    if(bellman() == false)
    {
        printf("Ciclu negativ!");
    }
    else
    {
        for(int i = 1; i < n; i++)
        {
            printf("%d ", dist[i]);
        }
    }

    return 0;
}