Cod sursa(job #1695064)

Utilizator ajeccAjechiloae Eugen ajecc Data 26 aprilie 2016 15:26:23
Problema Algoritmul Bellman-Ford Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 2.59 kb
#include <bits/stdc++.h>
#define for0(i,n) for(int i=0; i<n; i++)
#define for1(i,n) for(int i=1; i<=n; i++)
#define pb push_back
#define mp make_pair
#define all(v) v.begin(), v.end()
#define V vector<int>
#define VP vector<pair<int, int> >
#define clr(A,x) memset(A, x, sizeof(A))
#define cpy(A,B) memcpy(A, B, sizeof(B))
#define g(s) getline(cin, s) ///ai grija la fin/cin ///
#define FASTIO ios_base::sync_with_stdio(0)
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
/*template <typename T>
string to_string(const T& n){
    ostringstream os;
    os << n;
    return os.str();
}
*/

/*struct coord
{
    int x,y;

};
bool operator<(const coord &l, const coord &r)
{
    return (l.x<r.x || (l.x==r.x && l.y<r.y));
}*/

/*void invers_modular(int a, int b, int& d, int& x, int & y)
{
    if(!b)
    {
        d=a;
        x=1;
        y=0;
        return ;
    }
    int x0, y0;
    invers_modular(b, a%b, d, x0, y0);
    x=y0;
    y=x0-a/b*y0;
} // daca x<0 se aduna cu mod pana e mai mare, x fiind rezultatul*/

/*ull p(int baze, int exponent)
{
    if(exponent==0)
        return 1;

    if(exponent%2==0)
        return p(baze*baze, exponent/2);
    else return baze*p(baze, exponent-1);
}*/
ifstream fin("bellmanford.in"); /// modifica cu numele corespunzator
ofstream fout("bellmanford.out"); /// modifica cu numele corespunzator

int n, m;
int a, b, c;
unordered_map<int, int> drum_nod;
vector<vector<pair<int,int> > > lista(50001);

void citire();
void bellman();
void afisare();

bool ciclu=0;

int main()
{
    citire();
    bellman();
    afisare();

    return 0;
}

void citire()
{
    cin>>n>>m;
    for1(i,m)
    {
        cin>>a>>b>>c;
        lista[a].pb(mp(b,c));
        if(i<=n)
            drum_nod[i]=2100000000;
    }
    drum_nod[1]=0;
}

void bellman()
{
    int verificam_ciclu[50001];
    queue<int> coada;
    coada.push(1);
    while(!coada.empty())
    {
        int actual=coada.front();
        coada.pop();

        verificam_ciclu[actual]++;
        if(verificam_ciclu[actual]>n)
        {
            ciclu=1;
            return ;
        }

        for(auto i: lista[actual])
            if(drum_nod[i.first]>drum_nod[actual]+i.second)
            {
                drum_nod[i.first]=drum_nod[actual]+i.second;
                coada.push(i.first);
            }
    }

}

void afisare()
{
    if(ciclu)
    {
        cout<<"Ciclu negativ!";
        return ;
    }
    for(int i=2; i<=n; i++)
        cout<<(drum_nod[i]==2100000000?0:drum_nod[i])<<' ';
}