Cod sursa(job #3160154)

Utilizator AlexandraTomaToma Alexandra AlexandraToma Data 23 octombrie 2023 09:31:14
Problema BFS - Parcurgere in latime Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.49 kb
#include <iostream>
#include <cmath>
#include <fstream>
#include <vector>
#include <queue>
using namespace std;

//declarari globale
const int NMAX = 100000;
int vis[NMAX + 1] = {0}, d[NMAX + 1]; //vector vizitate, distante
vector <int> G[NMAX + 1]; //vector de vectori = lista de adiacenta

void BFS(int x) {
    queue <int> q;
    // adaugam nodul in coada
    q.push(x);
    // distanta intre x si x e 0
    d[x] = 0;
    // marcam ca vizitat
    vis[x] = 1;
    // parcurgem coada
    while (!q.empty()) {
        // preluam primul nod
        x = q.front();
        q.pop();
        // parcurgem lista de adiacenta pentru nodul curent
        for(auto next:G[x]) {
            // daca nu e inca vizitat
            if(!vis[next]) {
                //il punem in coada, il vizitam si ii retinem distanta
                q.push(next);
                vis[next] = 1;
                d[next] = d[x]+1;
            }
        }
    }
}

int main() {

    //fisiere I/O
    ifstream f("bfs.in");
    ofstream g("bfs.out");

    //citim graful
    int N, M, S;
    f >> N >> M >> S;
    for (int i = 1; i <= M ; ++i) {
        int x, y;
        f >> x >> y;
        G[x].push_back(y);
    }

    // facem BFS din nodul cerut
    BFS(S);

    //afisam distantele
    for(int i = 1; i <= N; i++) {
        // daca in parcurgere nu a fost vizitat nodul i, nu se poate ajunge din S la el
        if(vis[i] == 0)
            g << -1 << " ";
        else
            g << d[i] << " ";
    }

    f.close();
    g.close();
    return 0;
}