Cod sursa(job #887679)

Utilizator auRSTARHreapca Aurelian auRSTAR Data 24 februarie 2013 00:02:29
Problema Cerere Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.87 kb
#include<cstdio>
#include<vector>
#include<bitset>
using namespace std;
bitset<100010> suprem,viz;
vector<int> fiu[100010];
int i,n,x,y,A[100010],dad[100010],ROOT,cnt,ST[100010],SOL[100010],REZ[100010];
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 ",&A[i]);
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        suprem[y]=1;
        fiu[x].push_back(y);
        dad[y]=x;
    }
    for(i=1;i<=n;i++)if(!suprem[i])ROOT=i;
    REZ[ROOT]=0;
    DFS(ROOT);
    for(i=1;i<=n;i++)printf("%d ",REZ[i]);
    return 0;
}
void DFS(int X)
{
    ST[++cnt]=X;
    SOL[X]=ST[cnt-A[X]];
    if(SOL[X]!=X)REZ[X]=REZ[SOL[X]]+1;
    for(vector<int>::iterator it=fiu[X].begin();it!=fiu[X].end();it++)
        if(!viz[*it])DFS(*it);
    ST[cnt--]=0;
}