Cod sursa(job #1807393)

Utilizator rares00Foica Rares rares00 Data 16 noiembrie 2016 14:53:29
Problema Algoritmul Bellman-Ford Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 5.75 kb
//Bellman-Ford, graf orientat
#include <iostream>
#include <fstream>
#define INF 9999
using namespace std;
ifstream in("bellmanford.in");
ofstream out("bellmanford.out");

int n,m;

struct muchie{
    int x,y,cost;
}v[250005]; //vector muchii

int d[50005]; //vector distante minime st->i
int c[2500005]; //coada noduri vizitate
int viz[250005];   //vector noduri vizitate (care se afla in coada)

struct vecin{
    //inform utila
    int nod, cost;
    //inform de legatura
    struct vecin *urm;
}*L[50005],*act;

void citire()
{
    in>>n>>m;
    for(int i=1;i<=m;++i)
        in>>v[i].x>>v[i].y>>v[i].cost;
}

void citire1()
{
    int x,y,cost;
    in>>n>>m;
    for(int i=1;i<=m;++i)
    {
        in>>x>>y>>cost;
        //adauga vecinul y la lista x
        act=new vecin;
        act->nod=y;
        act->cost=cost;
        act->urm=L[x];
        L[x]=act;
    }
}

void bellmanFord(int st)
//drum distanta minima (fara coada)
{
    //initilizeaza vectorul distante cu +INF
    for(int i=1;i<=n;++i)
        d[i]=+INF;
    d[st]=0; //distanta st->st = 0

    for(int k=1;k<=n-1;++k) //n-1 iteratii
    {
        for(int i=1;i<=n;++i) //ia muchiile exterioare ale fiecarui nod
        {
            for(int j=1;j<=m;++j)   //parcurge vectorul de muchii
            {
                if(v[j].x==i)  //daca gasesti o muchie exterioara SI obtii un drum st->fin(al) mai mic trecand prin int(ermediar)
                    if(d[v[j].y] > d[i] + v[j].cost)    //daca( st->fin > st->int + int->fin )
                        d[v[j].y] = d[i] + v[j].cost;
            }
        }
    }

    for(int i=2;i<=n;++i)
        out<<d[i]<<" ";
    out<<"\n";
}

void bellmanFord1(int st)
//drum distanta minima (cu coada si vector de muchii)
{
    //initializez vectorul distante cu +INF
    for(int i=1;i<=n;++i)
        d[i]=+INF;
    d[st]=0;    //distanta st->st = 0

    //declarari
    int x,inc,sf;
    int cnt=1;

    inc=sf=1;
    c[inc]=st;
    viz[st]=1;


    while(inc<=sf) //cat timp coada nu e vida ?SI nu am facut inca n-1 iteratii
    {
        x=c[inc++]; //ia urmatorul nod din coada (ca nod intermediar)
        viz[x] = 0; //SI marcheaza nodul x ca nevizitat

        for(int i=1;i<=m;++i) //parcurge vectorul de muchii
        {
            if(v[i].x==x) //daca gasesti o muchie exterioara x->nod
                if(d[v[i].y] > d[x] + v[i].cost)  //daca obtii un drum optim (st->fin > st->int + int->fin)
                {
                    //actualizezi
                    d[v[i].y] = d[x] + v[i].cost;
                    //daca nodul final e nevizitat (nu se afla in coada)
                    if(viz[v[i].y] == 0)
                    {
                        c[++sf] = v[i].y;   //adaugi nodul final in coada
                        viz[v[i].y] = 1;    //SI marcheaza nodul ca vizitat
                    }
                }
        }
        ++cnt;
    }

    //afiseaza vectorul distante
    for(int i=2;i<=n;++i)
        out<<d[i]<<" ";
    out<<"\n";
}

bool bellmanFord2(int st)
//drum distanta minima (cu coada si vector de muchii)
{
    //initializez vectorul distante cu +INF
    for(int i=1;i<=n;++i)
        d[i]=+INF;
    d[st]=0;    //distanta st->st = 0

    //declarari
    int x,inc,sf;

    inc=sf=1;
    c[inc]=st;
    viz[st]++;


    while(inc<=sf) //cat timp coada nu e vida ?SI nu am facut inca n-1 iteratii
    {
        x=c[inc++]; //ia urmatorul nod din coada (ca nod intermediar)
         //SI marcheaza nodul x ca nevizitat

        for(int i=1;i<=m;++i) //parcurge vectorul de muchii
        {
            if(v[i].x==x) //daca gasesti o muchie exterioara x->nod
            {
                if(d[v[i].y]>d[x]+v[i].cost && viz[v[i].y]!=n-1)  //daca obtii un drum optim (st->fin > st->int + int->fin)
                {
                    //actualizezi
                    d[v[i].y] = d[x] + v[i].cost;
                        c[++sf] = v[i].y;   //adaugi nodul final in coada
                        viz[v[i].y] ++;    //SI marcheaza nodul ca vizitat
                }
                if(viz[v[i].y]==n-1)
                    return 0;
            }
        }
    }
    return 1;
}

bool bellmanFord3(int st)
//drum ditanta minima (cu coada si liste de adicenta alocate dinamic)
{
    //initializeaza vectorul distante cu +INF
    for(int i=1;i<=n;++i)
        d[i]=+INF;

    //initilizeaza vectorul noduri vizitate cu 0
//    for(int i=1;i<=n;++i)
//        viz[i]=0;

    //declarari
    int inc,sf,x,y;

    inc=sf=1;
    c[sf]=st;
    viz[st]++;
    d[st]=0;

    while(inc<=sf)  //cat timp coada nu e vida
    {
        //ia urmatorul nod din coada
        x = c[inc++];

        //ia toate muchiile exterioare ale nodului x
        act = L[x];
        while(act != NULL)
        {
            y = act->nod;
            //daca obtinem un nou drum optim st->y prin nodul x (st->y > st->x + x->y)
            if(d[y]>d[x]+act->cost && d[x]!=INF && viz[y]!=n-1)
            {
                d[y]=d[x]+act->cost;    //actualizeaza
                c[++sf]=y;  //adauga y in coada
                viz[y]++;
            }

            if(viz[y]==n-1)
                return 0;

            act=act->urm;
        }
    }

    return 1;
}

int main()
{
    //citire();
    citire1();

    //bellmanFord(1);
    //bellmanFord1(1);
/*
    bool ok = bellmanFord2(1);
    if(ok==0)
        out<<"Ciclu negativ!\n";
    else
    {
        for(int i=2;i<=n;++i)
            out<<d[i]<<" ";
        out<<"\n";
    }
*/
    bool ok = bellmanFord3(1);
    if(ok==0)
        out<<"Ciclu negativ!\n";
    else
    {
        for(int i=2;i<=n;++i)
            out<<d[i]<<" ";
        out<<"\n";
    }

    return 0;
}