Cod sursa(job #1085160)

Utilizator TeOOOVoina Teodora TeOOO Data 16 ianuarie 2014 21:12:32
Problema BFS - Parcurgere in latime Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.24 kb
//Include
#include <stdio.h>
#include <vector>
#include <queue>
using namespace std;

FILE *in, *out;

//Definitii
#define pb push_back

//COnstante
const int sz = (int)1e5+1;
const int oo = (1<<30)-1;

//Variabile
int nodes, edges, startNode;
int dist[sz];
vector <int> graph[sz];
queue <int> bfsQ;

//Main
int main()
{
    in = fopen("bfs.in","rt");
    out = fopen("bfs.out","wt");
    fscanf(in,"%d%d%d", &nodes, &edges, &startNode);
    while(edges--)
    {
        int rfrom, rto;
        fscanf(in,"%d%d", &rfrom, &rto);
        graph[rfrom].pb(rto);
    }
    for(int i=1; i<=nodes; ++i)
        dist[i] = oo;
    bfsQ.push(startNode);
    dist[startNode] = 0;

    while(!bfsQ.empty())
    {
        int currentNode = bfsQ.front();
        bfsQ.pop();

        vector<int>::iterator it, end = graph[currentNode].end();
        for(it = graph[currentNode].begin(); it!=end; ++it)
        {
            if(dist[currentNode]+1 < dist[*it])
            {
                dist[*it] = dist[currentNode]+1;
                bfsQ.push(*it);
            }
        }
    }

    for(int i=1; i<=nodes; ++i)
        fprintf(out, "%d ", dist[i] == oo? -1 : dist[i] );

    fclose(in);
    fclose(out);
    return 0;
}