Cod sursa(job #2011474)

Utilizator GeoeyMexicanuBadita George GeoeyMexicanu Data 16 august 2017 13:13:35
Problema Cerere Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.6 kb
#include <iostream>
#include <fstream>
#include <vector>
#define N 100010

using namespace std;

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

int arb[N],i,j,n,m,v[N],nr,k,r;
int dfs(int x)
{
    int p,z,ind,sol=0;
    while(v[x]!=0)
    {
        z=0;
        p=v[x];
        while(z<p)
        {
            x=arb[x];
            z++;
        }
        sol++;
    }
    return sol;
}
int main()
{
    f>>n;
    for(i=1;i<=n;i++)
        f>>v[i];
    for(i=1;i<n;i++)
    {
        f>>m>>k;
        arb[k]=m;
    }
    for(i=1;i<=n;i++){
        g<<dfs(i)<<' ';
    }
}