Cod sursa(job #3313297)

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

int dp[200005];
int st[200005];
vector<int>adj[200005];
bool vis[200005];
int k[200005];
void dfs(int node, int l)
{
    vis[node]=1;
    st[l]=node;
    dp[node]=dp[st[l-k[node]]]+1;
    for(auto i : adj[node])
    {
        if(vis[i]==0)
            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);
        adj[b].push_back(a);
    }
    dfs(poz, 1);
    for(int i=1; i<=n; i++)
        cout << dp[i]-1 << ' ';
    return 0;
}