Cod sursa(job #3362089)

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

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

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

void dp(int poz) {
    if (X[poz]!=0 && R[poz]==-1) {
        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 if (R[poz]==-1) R[poz]=0;
}

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