Cod sursa(job #1326188)

Utilizator gapdanPopescu George gapdan Data 24 ianuarie 2015 21:04:07
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.8 kb
#include<cstdio>
#include<vector>
#define NMAX 100005
using namespace std;
int a[NMAX],sol[NMAX];
int viz[NMAX],str[NMAX],st[NMAX];
int i,x,y,n,k,start;
vector<int>v[NMAX];
void DFS(int nod)
{
    viz[nod]=1;
    st[++k]=nod;
    if (a[nod]!=0) sol[nod]=sol[st[k-a[nod]]]+1;
    for (int i=0;i<v[nod].size();++i)
    {
        int it=v[nod][i];
        if (viz[it]==0) DFS(it);
    }
    --k;
}
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-1;++i)
    {
        scanf("%d%d",&x,&y);
        v[x].push_back(y);
        str[y]=1;
    }
    for (i=1;i<=n;++i)
        if (str[i]==0) start=i;
    DFS(start);
    for(i=1;i<=n;++i) printf("%d ",sol[i]);
    return 0;
}