Cod sursa(job #2069423)

Utilizator edi_laitinLaitin Eduard edi_laitin Data 18 noiembrie 2017 13:34:10
Problema Cerere Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.67 kb
#include <iostream>
#include <fstream>

using namespace std;

ifstream fin("cerere.in");
ofstream fout("cerere.out");

int N,K[100005],T[100005];

void Read()
{
    fin>>N;

    for(int i=1;i<=N;i++)
        fin>>K[i];

    for(int i=1;i<N;i++)
    {
        int a,b;

        fin>>a>>b;
        T[b]=a;
    }
}

void Solve()
{
    for(int i=1;i<=N;i++)
    {
      int q=i,nr=0;

       while(K[q]!=0)
       {
           int x=K[q];

           while(x)
           {
              q=T[q];
              x--;
           }
           nr++;
       }
       fout<<nr<<" ";
    }
}

int main()
{
    Read();
    Solve();
    return 0;
}