Cod sursa(job #2663982)

Utilizator CostinteoGrigore Costin Teodor Costinteo Data 27 octombrie 2020 18:30:48
Problema BFS - Parcurgere in latime Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.11 kb
#include <fstream>
#include <iostream>
#include <vector>
#include <iterator>
#include <queue>

#define NMAX 1000001

std::vector <int> adj[NMAX];
int visited[NMAX], distance[NMAX];
int q[NMAX], left = 0, right = 0;

void read(int &n, int &node) {
    std::ifstream f("bfs.in");
    f >> n;
    int edges;
    f >> edges;
    f >> node;
    for (int i = 0; i < edges; i++) {
        int x, y;
        f >> x >> y;
        adj[x - 1].push_back(y - 1);
    }
    f.close();
}

void bf_search(int x, int n) {

    for (int i = 0; i < n; i ++) {
        visited[i] = 0;
        distance[i] = -1;
    }

    distance[x] = 0;
    q[right] = x;
    right++;
    while (left <= right) {
        visited[q[left]] = 1;
        for (int to_node : adj[q[left]]) {
            if (!visited[to_node]) {
                q[right++] = to_node;
                distance[to_node] = distance[q[left]] + 1;
            }
        }
        left++;
    }

    std::ofstream g("bfs.out");

    for (int i = 0; i < n; i++) {
        g << distance[i] << " ";
    }

    g.close();
}

int main() {
    int n, x;
    read(n, x);
    bf_search(x - 1, n);
    return 0;
}