Cod sursa(job #3230037)

Utilizator Sabin1133Padurariu Sabin Sabin1133 Data 18 mai 2024 21:18:04
Problema BFS - Parcurgere in latime Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.26 kb
#include <fstream>
#include <queue>
#include <vector>

#define NMAX 100000

#define INFINITY (1<<30)

#define VISITED 1
#define NOTVISITED 0


std::vector<int> adj[NMAX];
int dist[NMAX];
int vis[NMAX];


void bfs(int source, std::vector<int> adj[NMAX], int n, int dist[])
{
    std::queue<int> q;

    for (int i = 0; i < n; ++i) {
        vis[i] = NOTVISITED;
        dist[i] = INFINITY;
    }

    dist[source] = 0;
    vis[source] = VISITED;

    q.push(source);

    while (!q.empty()) {
        auto node = q.front();
        q.pop();

        for (auto neigh : adj[node]) {
            if (!vis[neigh]) {
                dist[neigh] = dist[node] + 1;
                q.push(neigh);
            }
        }
    }
}

int main()
{
    std::ifstream fin("bfs.in");
    std::ofstream fout("bfs.out");

    int n, m, source;


    fin >> n >> m >> source;

    --source;

    for (int v, w, i = 0; i < m; ++i) {
        fin >> v >> w;
        --v;
        --w;
        adj[v].push_back(w);
        adj[w].push_back(v);
    }


    bfs(source, adj, n, dist);


    for (int i = 0; i < n; ++i) {
        if (dist[i] == INFINITY)
            fout << -1 << '\n';
        else
            fout << dist[i] << '\n';
    }

    fout << '\n';

    fin.close();
    fout.close();

    return 0;
}