Cod sursa(job #2116089)

Utilizator patcasrarespatcas rares danut patcasrares Data 27 ianuarie 2018 12:19:27
Problema Cerere Scor 85
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.01 kb
#include<cstdio>
#include<vector>
#include<cmath>
#include<queue>
#define DN 100005
#define pb push_back
using namespace std;
int n,p,q,dp[DN],k[DN],pr[25][DN],d,f,z[DN],niv[DN],l;
vector<int>v[DN];
void dfs(int nod)
{
    z[niv[nod]]=nod;
    if(k[nod]!=0)
    {
        f=k[nod];
        l=niv[nod]-k[nod];
        d=z[l];
        dp[nod]=dp[d]+1;
    }
    for(auto i:v[nod])
    {
        niv[i]=niv[nod]+1;
        dfs(i);
    }
}
int main()
{
    freopen("cerere.in","r",stdin);
    freopen("cerere.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&k[i]);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&p,&q);
        pr[0][q]=p;
        v[p].pb(q);
    }
    for(int j=1;j<17;j++)
        for(int i=1;i<=n;i++)
            pr[j][i]=pr[j-1][pr[j-1][i]];
    for(int i=1;i<=n;i++)
        if(pr[0][i]==0)
        {
            niv[i]=1;
            dfs(i);
            break;
        }
    for(int i=1;i<=n;i++)
        printf("%d ",dp[i]);
}