Cod sursa(job #2667736)

Utilizator SahisttulArsene Marinel Sahisttul Data 3 noiembrie 2020 19:41:20
Problema BFS - Parcurgere in latime Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.61 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream f("bfs.in");
ofstream g("bfs.out");

const int NMAX = 1000001;

int distanta[NMAX];                /// vector de muchii vizitate unde salvam distantele
int Coada[NMAX], n, m, s;
vector<int> arce[NMAX];            /// vector pentru lista de muchii


void citire()
{
    int x, y;
    f >> n >> m >> s;
    for(int i = 0; i < m; i++)
    {
        f >> x >> y;
        arce[x].push_back(y);
    }
}


void BFS()
{
    int st = 0, dr = 0;

    distanta[s] = 0;      ///distanta nodului din care plecam este 0
    Coada[0] = s;         ///adaugam nodul in coada

    while(st <= dr)      ///parcurgem graful
    {
        int x = Coada[st++];   ///extragem nodul curent din coada
        int aux = arce[x].size();
        for (int i = 0; i < aux; i++)      ///parcurgem vectorul de arce al nodului curent (x)
        {
            int vecin_curent = arce[x][i];
            if (distanta[vecin_curent] == -1)                ///daca nodul este nevizitat
            {
                distanta[vecin_curent] = distanta[x] + 1;       ///distanta drumului devine distanta parcursa pana acum + 1
                Coada[++dr] = vecin_curent;              ///adaugam noul nod gasit in coada
            }
        }
    }
}

void afis()
{
    for(int i = 1; i <= n; i++)
        g << distanta[i] << ' ';
}


int main()
{
    citire();
    //
    for(int i = 1; i <= n; i++)
        distanta[i] = -1;            /// initializam distanta fiecarui nod cu -1
    //
    BFS();
    //
    afis();
    return 0;
}