Cod sursa(job #2073616)

Utilizator GeorgeCalinPetruta George-Calin GeorgeCalin Data 23 noiembrie 2017 13:48:43
Problema Cerere Scor 25
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.24 kb
#include <fstream>
#include <vector>
#include <queue>
#define nmax 100002
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");

int pas[nmax];
int sol[nmax];
vector <int> graf[nmax];
queue <int> coada;
int v2[nmax];
int viz[nmax];

void bfs()
{
    while(!coada.empty())
    {
        int nod=coada.front();
        coada.pop();
        for(int i=0;i<graf[nod].size();i++)
        {
            if(!viz[graf[nod][i]])
            {
                viz[graf[nod][i]]=1;
                v2[graf[nod][i]]=v2[nod]+1;
                coada.push(graf[nod][i]);
            }
        }
    }
}

int main()
{
    int n,x,y;
    fin>>n;
    for(int i=1;i<=n;i++)
    {
        fin>>pas[i];
        if(pas[i]==0){
            coada.push(i);
            viz[i]=1;
        }
    }
    for(int i=1;i<=n;i++)
    {
        fin>>x>>y;
        graf[x].push_back(y);
    }
    bfs();
    for(int i=1;i<=n;i++)
    {
        if(v2[i]==0)
        {
            fout<<0;
        }
        else
        {
            if(v2[i]<pas[i])
            {
                fout<<1;
            }
            else
                fout<<v2[i]-pas[i]+1;
        }
        fout<<" ";
    }
    return 0;
}