Cod sursa(job #2723561)

Utilizator Florian11232Florian Susai Florian11232 Data 14 martie 2021 20:20:54
Problema BFS - Parcurgere in latime Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.88 kb
#include <iostream>
#include <vector>
#include <fstream>
#include <queue>
using namespace std;

// numarul maxim de noduri
const int Nmax = 100005;

// distanta maxima pe care o poate avea un nod
const int INF = 100001;

vector<int> graph[Nmax];

queue <int> q;
int d[Nmax];

ifstream in("bfs.in");
ofstream out("bfs.out");

int main()
{
    int N, M, S, x, y;

    in >> N; // numarul de noduri
    in >> M; // numarul de arce
    in >> S; // indicele nodului de start

    for (int i = 0; i < M; i++) {
        in >> x >> y;
        // graf orientat
        graph[x].push_back(y); // adaug in lista vecinilor lui x pe y
    }

    // afisare lista vecini
    // for (int node = 1; node <= N; node++) {
    //     cout << node << ": ";
    //     // parcurg toti vecinii lui node
    //     for (int y = 0; y < graph[node].size(); y++) {
    //         cout << graph[node][y] << " ";
    //     }
    //     cout << '\n';
    // }

    // initializare vector de distante (d) cu INF
    for (int i = 1; i <= N; i++) {
        d[i] = INF;
    }

    // adaug nodul de start in coada
    q.push(S);
    // marchez nodul de start cu valoarea 0
    d[S] = 0;

    // cat timp coada este nevida
    while(!q.empty()) {
        // node <- elementul de la inceputul cozii
        int node = q.front();
        
        // elimin nodul de la inceputul cozii
        q.pop();

        // parcurg toti vecinii y ai lui node
        for (int i = 0; i < graph[node].size(); i++) {
            y = graph[node][i];

            if (d[y] > 1 + d[node]) {
                d[y] = 1 + d[node];
                // adaug in coada nodul y
                q.push(y);
            }
        }
    }

    for (int i = 1; i <= N; i++) {
        if (d[i] == INF) {
            d[i] = -1;
        }
        out << d[i] << " ";
    }
    out << '\n';

    return 0;

}