Cod sursa(job #3236271)

Utilizator Dia3141Costea Diana Stefania Dia3141 Data 27 iunie 2024 11:26:27
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.7 kb
#include <fstream>
#include <vector>
#define dim 100002
using namespace std;
ifstream cin("cerere.in");
ofstream cout("cerere.out");
int n,x,y,t[dim],stramos[dim],dp[dim],cost[dim],root,k;
vector<int>v[dim];
void dfs(int x){
    stramos[++k]=x;
    dp[x]=dp[stramos[k-cost[x]]]+1;
    for(auto i: v[x])
        dfs(i);
    k--;
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>cost[i];
    for(int i=1;i<n;i++){
        cin>>x>>y;
        t[y]=x;
        v[x].push_back(y);
    }
    for(int i=1;i<=n;i++)
        if(t[i]==0){
            root=i;
            break;
        }
    dfs(root);
    for(int i=1;i<=n;i++)
        cout<<dp[i]-1<<" ";
    return 0;
}