Cod sursa(job #3162254)

Utilizator robertapopescuPopescu Roberta Andreea robertapopescu Data 28 octombrie 2023 18:17:36
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.93 kb
#include <bits/stdc++.h>

using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
const int N=1e5+10;
vector <int> a[N];
int vf[2*N],urm[2*N],lst[N],nr;
void adauga(int x,int y)
{
    vf[++nr]=y;
    urm[nr]=lst[x];
    lst[x]=nr;
}
int tata[N],n,val[N],rez[N],stramos[N];
void calc(int x,int dist)
{
    stramos[dist]=x;
    rez[x]=rez[stramos[dist-val[x]]]+1;
    for(int t=lst[x];t!=0;t=urm[t])
    {
        int i=vf[t];
        calc(i,dist+1);
    }
}
int main()
{
    ios_base::sync_with_stdio(false);
    f>>n;
    for(int i=1;i<=n;i++)
    {
        f>>val[i];
        rez[i]=-1;
    }
    for(int i=1;i<n;i++)
    {
        int x,y;
        f>>x>>y;
        tata[y]=x;
        adauga(x,y);
    }
    for(int i=1;i<=n;i++)
    {
        if(tata[i]==0)
        {
            calc(i,0);
        }
    }
    for(int i=1;i<=n;i++)
    {
        g<<rez[i]<<" ";
    }
    return 0;
}