Cod sursa(job #1177293)

Utilizator andi23roscaRosca Andrei andi23rosca Data 26 aprilie 2014 12:12:43
Problema BFS - Parcurgere in latime Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.25 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>

#define maxn 100010

using namespace std;

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

vector<int> muchii[maxn];

queue<int> coada;

int vizitati[maxn], cost[maxn], ok=0;

void citire(int &n, int &m, int &s)
{
    int i,x,y;

    f>>n>>m>>s;

    for(i=1;i<=m;i++)
    {
        f>>x>>y;

        muchii[x].push_back(y);

        if(x==s && y==s)
                ok=1;
    }
}

int main()
{
    int n,m,s,i,distanta=0,nod;

    vector<int>::iterator it;

    citire(n,m,s);
    coada.push(s);

    while(!coada.empty())
    {
        nod = coada.front();
        vizitati[nod] = 1;
        distanta = cost[nod] + 1;
        coada.pop();

        for(it = muchii[nod].begin(); it != muchii[nod].end(); it++)
        {
            if(vizitati[*it] != 1)
            {
                coada.push(*it);
                cost[*it] = distanta;
            }
        }

    }

    for(i=1;i<=n;i++)
    {
       if(cost[i]==0)
       {
           if(i==s)
           {
               if(ok==0) g<<"-1 ";
               else g<<"0 ";
           }
           else g<<"-1 ";
       }
       else g<<cost[i]<<" ";
    }


    return 0;
}