Cod sursa(job #1897428)

Utilizator MithrilBratu Andrei Mithril Data 1 martie 2017 13:46:49
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
const int NMAX = 100005;
int n,x,y;
int kcost[NMAX];
int dp[20][NMAX];

int getParent(int nod,int wow)
{
    if(memo[i]!=-1)
    int l=kcost[nod];
    for(int p=0 ; (1<<p)<=l ; p++)
        if(l&(1<<p))
            nod=dp[p][nod];
    if(kcost[nod]) return getParent(nod,wow+1);
    return wow;
}

int main()
{
    fin>>n;
    for(int i=1;i<=n;i+=1) fin>>kcost[i];
    for(int i=1;i<n;i+=1)
    {
        fin>>x>>y;
        dp[0][y]=x;
    }

    for(int p=1; (1<<p)<=n ; p++)
    for(int i=1; i<=n; i++)
        dp[p][i]=dp[p-1][dp[p-1][i]];

    for(int i=1;i<=n;i+=1)
    {
        if(kcost[i]==0) fout<<0<<' ';
        else fout << getParent(i,1)<<' ';
    }
    return 0;
}