Cod sursa(job #2914160)

Utilizator bumblebeeGeorge Bondar bumblebee Data 19 iulie 2022 08:22:10
Problema BFS - Parcurgere in latime Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.2 kb
#include <fstream>
#include <queue>
#include <vector>
#include <cstring>
#define MAX_VERTICES 100010
using namespace std;
 
ifstream fin("bfs.in");
ofstream fout("bfs.out");

vector<int> adj[MAX_VERTICES];
 
void compute_distances(int start_vertex, int nr_vertices) {
    queue<int> vertices;
    int distances[MAX_VERTICES];
    memset(distances, -1, sizeof(distances));
    vertices.push(start_vertex);
    distances[start_vertex] = 0;
    while (!vertices.empty()) {
        int current = vertices.front();
        vertices.pop();
        for (int &neighbor : adj[current]) {
            if (distances[neighbor] == -1) {
                vertices.push(neighbor);
                distances[neighbor] = distances[current] + 1;
            }
        }
    }
    for (int i = 1; i <= nr_vertices; ++i) {
        fout << distances[i] << " ";
    }
    fout << '\n';
}

int main() {
    int nr_vertices, nr_edges, start_vertex, vertex_a, vertex_b;
    fin >> nr_vertices >> nr_edges >> start_vertex;
    for (int i = 0; i < nr_edges; ++i) {
        fin >> vertex_a >> vertex_b;
        adj[vertex_a].push_back(vertex_b);
    }
    compute_distances(start_vertex, nr_vertices);
    return 0;
}