Cod sursa(job #2238733)

Utilizator Andreea_DanielaAndreea Guster Andreea_Daniela Data 7 septembrie 2018 12:27:42
Problema BFS - Parcurgere in latime Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.37 kb
#include <iostream>
#include <fstream>
#include <queue>
using namespace std;

void lant ( int y, int* tata, int &nr_arce)
{
    if(y!=0)
        lant(tata[y], tata, nr_arce);
    nr_arce++;

}

void BFS(int sursa, const vector<vector<int>> &la, int n, ofstream &fout)
{
    int *viz = new int[n+1];
    int *tata = new int[n+1];
    queue<int> c;

    for(int i=1; i<=n; i++)
        viz[i] = tata[i] = 0;

    viz[sursa] = 1;
    c.push(sursa);

    while(!c.empty())
    {
        int u = c.front();
        c.pop();

        for(unsigned i=0; i< la[u].size(); i++)
        {
            int v = la[u][i];
            if(!viz[v])
            {
                tata[v] = u;
                viz[v] = 1;
                c.push(v);
            }
        }
    }


    for(int i=1; i<=n; i++)
    {   int nr_arce = -2;
        if( !(tata[i] == 0 && i != sursa ) ){
          lant(i, tata, nr_arce);
          fout << nr_arce << " ";
        }
        else
            fout << -1 << " ";
    }
}

int main()
{
    ifstream fin ("bfs.in");
    ofstream fout ("bfs.out");

    int n, m, s;
    fin >> n >> m >> s;

    vector<vector<int>> la(n+1);
    for(int i=1; i<=m; i++)
    {
        int x, y;
        fin >> x >> y;
        la[x].push_back(y);
    }
    fin.close();

    BFS(s, la, n, fout);

    fout.close();
    return 0;
}