Cod sursa(job #1234350)

Utilizator CosminRusuCosmin Rusu CosminRusu Data 27 septembrie 2014 11:07:24
Problema BFS - Parcurgere in latime Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.81 kb
#include <fstream>
#include <iostream>
#include <vector>
#include <bitset>
#include <string.h>
#include <algorithm>
#include <iomanip>
#include <math.h>
#include <time.h>
#include <stdlib.h>
#include <set>
#include <map>
#include <string>
#include <queue>
#include <deque>

using namespace std;

const char infile[] = "bfs.in";
const char outfile[] = "bfs.out";

ifstream fin(infile);
ofstream fout(outfile);

const int MAXN = 100005;
const int oo = 0x3f3f3f3f;

typedef vector<int> Graph[MAXN];
typedef vector<int> :: iterator It;

const inline int min(const int &a, const int &b) { if( a > b ) return b;   return a; }
const inline int max(const int &a, const int &b) { if( a < b ) return b;   return a; }
const inline void Get_min(int &a, const int b)    { if( a > b ) a = b; }
const inline void Get_max(int &a, const int b)    { if( a < b ) a = b; }

class DirectedGraph {
private:
    vector <vector <int> > G;
public:
    DirectedGraph() {
    }
    DirectedGraph(int _N) {
        G.resize(_N);
    }
    void addEdge(int x, int y) {
        G[x].push_back(y);
    }
    vector <int> bfs(int Source) {
        vector <int> dist(G.size(), -1);
        queue <int> Q;
        Q.push(Source);
        dist[Source] = 0;
        while(!Q.empty()) {
            int node = Q.front();
            Q.pop();
            for(auto it: G[node])
                if(dist[it] == -1) {
                    dist[it] = dist[node] + 1;
                    Q.push(it);
                }
        }
        return dist;
    }
};

int N, M, Source;

int main() {
    fin >> N >> M >> Source;
    DirectedGraph G(N);
    for( ; M -- ; ) {
        int x, y;
        fin >> x >> y;
        -- x ; -- y;
        G.addEdge(x, y);
    }
    vector <int> ans = G.bfs(Source - 1);
    for(auto it:ans)
        fout << it << ' ';
    fin.close();
    fout.close();
    return 0;
}