Cod sursa(job #3362085)

Utilizator Tudor_TapeTapu Tudor Cristian Tudor_Tape Data 2 august 2026 12:24:33
Problema Cerere Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.78 kb
#include <fstream>
using namespace std;

ifstream cin("cerere.in");
ofstream cout("cerere.out");

struct point {
    int p;
};
point P[100005];
int X[100005];
bool F[100005];
int R[100005];

void dp(int poz) {
    if (X[poz]!=0) {
        int val=X[poz], mpoz=poz;
        for (int i=1; i<=val; i++) poz=P[poz].p;
        dp(poz);
        R[mpoz]=R[poz]+1;
    }
    else R[poz]=0;
}

int main() {
    int n;
    cin>>n;
    for (int i=1; i<=n; i++) cin>>X[i];
    for (int i=1; i<=n-1; i++) {
        int x, y;
        cin>>x>>y;
        F[y]=1;
        P[y].p=x;
    }
    for (int i=1; i<=n; i++) {
        R[i]=-1;
    }
    for (int i=1; i<=n; i++) {
        if (R[i]==-1) dp(i);
    }
    for (int i=1; i<=n; i++) {
        cout<<R[i]<<" ";
    }
}