Cod sursa(job #2632119)

Utilizator loraclorac lorac lorac Data 2 iulie 2020 11:54:01
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream cin("cerere.in");
ofstream cout("cerere.out");
const int lim=1e5+4;
vector<int> vec[lim];
int deg[lim],drum[lim];
int k[lim],ans[lim];
void df(int nod,int depth)
{
    drum[depth]=nod;
    if(k[nod]==0) ans[nod]=0;
    else ans[nod]=ans[drum[depth-k[nod]]]+1;
    for(int x:vec[nod]) df(x,depth+1);
}
int main()
{
    int n,root,a,b;
    cin>>n;
    for(int i=1;i<=n;++i)
        cin>>k[i];
    for(int i=1;i<n;++i)
    {
        cin>>a>>b;
        vec[a].push_back(b);
        ++deg[b];
    }
    root=0;
    for(int i=1;i<=n and !root;++i)
    if(!deg[i]) root=i;
    df(root,1);
    for(int i=1;i<=n;++i)
        cout<<ans[i]<<' ';
    return 0;
}