Cod sursa(job #3351169)

Utilizator VladStroicaStroica Vlad Cristian VladStroica Data 17 aprilie 2026 13:06:54
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <bits/stdc++.h>
using namespace std;
int n;
int v[100005];
int v2[100005];
int v3[100005];
int rez[100005];
vector<int>adj[100005];
void Func(int poz)
{
    for(int i=0;i<adj[v3[poz]].size();i++)
    {
        //cout<<adj[v3[poz]][i]<<" ";
        if(v[adj[v3[poz]][i]]!=0)
        rez[adj[v3[poz]][i]]=rez[v3[poz-v[adj[v3[poz]][i]]+1]]+1;
        v3[poz+1]=adj[v3[poz]][i];
        Func(poz+1);
    }
}

int main()
{
    ifstream cin("cerere.in");
    ofstream cout("cerere.out");
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>v[i];
    }
    for(int i=1;i<n;i++)
    {
        int a,b;
        cin>>a>>b;
        adj[a].push_back(b);
        v2[b]++;
    }
    int rad=0;
    for(int i=1;i<=n;i++)
    {
        if(v2[i]==0)
        {
            rad=i;
            break;
        }
    }
    v3[1]=rad;
    Func(1);
    for(int i=1;i<=n;i++)
        cout<<rez[i]<<" ";

    return 0;
}