Cod sursa(job #1561446)

Utilizator geni950814Geni Geni geni950814 Data 4 ianuarie 2016 06:22:28
Problema BFS - Parcurgere in latime Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.98 kb
#include <fstream>
#include <deque>
#include <vector>
#include <iostream>

using namespace std;

int N, M, S;
vector<vector<int>> V;
vector<int> sol;
deque<int> dq;

int main() {

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

    in >> N >> M >> S;

    int fst, snd;
    
    for(int i = 0; i <= N; i++) {
        V.push_back(vector<int>(1, 0));
        sol.push_back(-1);
    }

    for(int i = 0; i < M; i++) {
        in >> fst >> snd;
        if(fst != snd && snd != S) {
            V[fst][0]++;
            V[fst].push_back(snd);
            //V[fst][++V[fst][0]] = snd;    
        }
    }
    
    dq.push_back(S);
    sol[S] = 0;

    while(!dq.empty()) {
        int curr = dq.front();
        dq.pop_front();
        int n = V[curr][0];
        for(int i = 1; i <= n; i++) {
            int e = V[curr][i];
            dq.push_back(e);
            sol[e] = sol[curr]+1;
        }
    }
   
    for(int i = 1; i <= N; i++) {
        out << sol[i] << " ";
    }

    return 0;
}