Cod sursa(job #1786854)

Utilizator Julian.FMI Caluian Iulian Julian. Data 23 octombrie 2016 19:26:00
Problema Cerere Scor 100
Compilator cpp Status done
Runda Teme Pregatire ACM Unibuc 2013 Marime 0.67 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <stack>
#define nmax 100099
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
vector <int> g[nmax];
int k[nmax],rez[nmax],s[nmax],sf[nmax];
char viz[nmax];

int main()
{int i,n,a,b,rad,fiu;
fin>>n;
for(i=1;i<=n;i++)fin>>k[i];
for(i=1;i<=n-1;i++)
    {fin>>a>>b;  viz[b]=1; g[a].push_back(b); }
rad=1;while(viz[rad])rad++;
//dfs:
int vf=1;
s[1]=rad;vf=1;sf[1]=0;
    while(vf)
    {rad=s[vf]; fiu=sf[vf];
     if(k[rad])rez[rad]=rez[s[vf-k[rad]]]+1;
     if(fiu<g[rad].size())
     {s[++vf]=g[rad][fiu]; sf[vf]=0; sf[vf-1]=fiu+1;
     }else vf--;
    }
for(i=1;i<=n;i++)fout<<rez[i]<<' ';

}