Cod sursa(job #2916419)

Utilizator AxicaVirtosu Alexandra Mihaela Axica Data 29 iulie 2022 17:31:46
Problema Cerere Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.7 kb

#include <fstream>
#include <vector>
#include <algorithm>
#define NMAX 100010
using namespace std;
ifstream fin ("cerere.in");
ofstream fout ("cerere.out");

int n;
int nrstramos[NMAX];
int tata[NMAX];
int main()
{
    fin>>n;
    for(int i=1; i<=n; i++)
        fin>>nrstramos[i];
    int x, y;
    for(int i=1; i<=n-1; i++)
    {
        fin>>x>>y;
        tata[y]=x;
    }

    for(int i=1; i<=n; i++)
    {
        int nr=0, x=i;///x este maimuta care rezolva prob
        while(nrstramos[x]!=0)
        {
            nr++;
            int aux=nrstramos[x];
            while(aux)
                {x=tata[x]; aux--;}
        }
        fout<<nr<<" ";
    }

    return 0;
}