Cod sursa(job #918738)

Utilizator iuli1505Parasca Iuliana iuli1505 Data 19 martie 2013 09:11:22
Problema Cerere Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.84 kb
#include<cstdio>
#include<vector>
#define nmax 100010
using namespace std;
vector<int>g[nmax];
int n,k[nmax],dad[nmax],i,sol[nmax],viz[nmax],prec[nmax],q[nmax],x,y,nr;
void dfs(int);
int main()
{
    freopen("cerere.in","r",stdin);
    freopen("cerere.out","w",stdout);
    scanf("%d", &n);
    for(i=1;i<=n;++i)scanf("%d", &k[i]);
    for(i=1;i<n;++i){scanf("%d%d", &x, &y);g[x].push_back(y);dad[y]=x;}
    for(i=1;i<=n&&dad[i];++i);
    dfs(i);
    for(i=1;i<=n;++i)
        printf("%d ", sol[i]);
    return 0;
}
void dfs(int maimuta)
{
    viz[maimuta]=1;
    q[++nr]=maimuta;
    prec[maimuta]=q[nr-k[maimuta]];
    if(prec[maimuta]!=maimuta)sol[maimuta]=sol[prec[maimuta]]+1;
    for(vector<int>::iterator it=g[maimuta].begin();it!=g[maimuta].end();++it)
        if(!viz[*it])
            dfs(*it);
    q[nr]=0;
    --nr;
}