Cod sursa(job #3313303)

Utilizator iustincmbMaican Iustin iustincmb Data 3 octombrie 2025 11:44:13
Problema Cerere Scor 65
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <bits/stdc++.h>
using namespace std;

int dp[200005];
int st[200005];
vector<int>adj[200005];
int k[200005];
void dfs(int node, int l)
{
    st[l]=node;
    if(k[node]!=0)
        dp[node]=dp[st[l-k[node]]]+1;
    else
        dp[node]=0;
    for(auto i : adj[node])
    {
        dfs(i, l+1);
    }
}
int main() 
{
    ifstream cin ("cerere.in");
    ofstream cout ("cerere.out");
    int n;
    cin >> n;
    int poz=-1;
    for(int i=1; i<=n; i++)
    {
        cin >> k[i];
        if(k[i]==0 && poz==-1)
            poz=i;
    }
    for(int i=1; i<=n-1; i++)
    {
        int a, b;
        cin >> a >> b;
        adj[a].push_back(b);
    }
    dfs(poz, 1);
    for(int i=1; i<=n; i++)
        cout << dp[i] << ' ';
    return 0;
}