Cod sursa(job #2651901)

Utilizator felix24mihaiPrava Felix Mihai felix24mihai Data 23 septembrie 2020 19:14:14
Problema Cerere Scor 50
Compilator cpp-32 Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <iostream>
#include <fstream>
#define NMAX 100003
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int n, ancestorsLvl[NMAX], father[NMAX];
void read(){
    int a,b;
    f >> n;
    for (int i = 0; i < n; i++){
        f >> ancestorsLvl[i+1];
        //g << ancestorsLvl[i+1];
    }
    for (int i = 1; i < n; i++){
        f >> a >> b;
        father[b] = a;
    }
}
int ancesLvl(int node, int lvl){
    for (int i = 0; i < lvl; i++)
        node = father[node];
    return node;
}
int resolve(int node, int nr){
    while (ancestorsLvl[node] != 0){
    	nr++;
    	node = ancesLvl(node,ancestorsLvl[node]);
    }
    return nr;
}
int main()
{
    read();
    for (int i = 1; i <= n; i++)
        g << resolve(i,0) << " ";
    return 0;
}