Cod sursa(job #2033075)

Utilizator zeboftwAlex Mocanu zeboftw Data 6 octombrie 2017 08:58:41
Problema BFS - Parcurgere in latime Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.76 kb
#include <fstream>
#include <queue>
#include <set>
#include <vector>
#include <cstring>

using namespace std;

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

const int N_MAX = 100000;
const int INF = 0x3f3f3f3f;

vector<int> AD[N_MAX];
int dist[N_MAX];

int main()
{
    int n, m, s;

    fin >> n >> m >> s;

    for (int i=1; i <= m; i++) {
        int from, to;
        fin >> from >> to;
        if (from != to)
        AD[from].push_back(to);
    }

    /*

    for (int i=1; i<=n; i++) {
        cout << " node " << i << " is adiacent to ";
        for (vector<int>::iterator it = AD[i].begin(); it != AD[i].end(); it++) {
            cout << *it << ' ';
        }
        cout << '\n';
    }

    */

    for (int i=1; i <= n; i++)
    {
        queue<int> coada;
        set<int> checked;
        int lung=0;
        bool ok=0;
        memset(dist, INF, sizeof(dist));
        dist[s]= 0;

        coada.push(s);
        checked.insert(s);

        while (!coada.empty())
        {
            int current = coada.front();
            coada.pop();
            if (current == i) {
                fout << dist[current] << ' ';
                ok = 1;
                break;
            }
            else {
                for (vector<int>::iterator it = AD[current].begin(); it != AD[current].end(); it++) {
                    int node = *it;
                    if (checked.find(node) == checked.end()) {
                        checked.insert(node);
                        coada.push(node);
                        if (dist[node] > dist[current] + 1) dist[node] = dist[current] + 1;
                    }
                }
            }
        }
        if (!ok) fout << "-1 ";
    }

    return 0;
}