Cod sursa(job #1897929)

Utilizator tqmiSzasz Tamas tqmi Data 1 martie 2017 19:16:53
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");

vector <int> G[100005];
int N,dp[100005],st[100005],nst,K[100005];


void Dfs(int nod)
{
    if(!K[nod])
        dp[nod]=0;
    else
        dp[nod]=dp[st[nst-K[nod]+1]]+1;
    st[++nst]=nod;
    for(int i=0;i<G[nod].size();i++)
    {
        int son=G[nod][i];
        Dfs(son);
    }
    nst--;
}

int main()
{
    fin>>N;
    for(int i=1;i<=N;i++)
    {
        fin>>K[i];
    }
    for(int i=1;i<N;i++)
    {
        int x,y;
        fin>>x>>y;
        G[x].push_back(y);
    }
    Dfs(1);
    for(int i=1;i<=N;i++)
    {
        fout<<dp[i]<<" ";
    }
    return 0;
}