Cod sursa(job #2724979)

Utilizator levladiatorDragutoiu Vlad-Ioan levladiator Data 18 martie 2021 11:03:52
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.04 kb
#include <bits/stdc++.h>
#define mp make_pair
#define pb push_back
#define pf push_front
#define ll long long
#define ull unsigned long long
#define x first
#define y second
#define pi pair<int,int>
#define pl pair<ll,ll>

using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");

const ll N=1e5+5,INF=1e18,MOD=666013,M=1e2+5,inf=INT_MAX;

int n;
int k[N],tata[N],ans[N],in[N];
vector<int> adj[N];

void dfs(int nod,int cnt)
{
    in[cnt]=nod;
    if(k[nod])ans[nod]=ans[in[cnt-k[nod]]]+1;
    for(int i=0;i<adj[nod].size();i++)
    {
        int vec=adj[nod][i];
        dfs(vec,cnt+1);
    }
}

int main()
{
    fin>>n;
    for(int i=1;i<=n;i++)
    {
        fin>>k[i];
    }
    for(int i=1;i<n;i++)
    {
        int a,b;
        fin>>a>>b;
        adj[a].pb(b);
        tata[b]=a;
    }
    for(int i=1;i<=n;i++)
    {
        if(tata[i]==0)
        {
            dfs(i,1);
            break;
        }
    }
    for(int i=1;i<=n;i++)
    {
        fout<<ans[i]<<" ";
    }
}