Cod sursa(job #1835279)

Utilizator Mircea_DonciuDonciu Mircea Mircea_Donciu Data 26 decembrie 2016 17:20:23
Problema Cerere Scor 95
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.85 kb
#include <fstream>
#include <vector>
#include <stack>
using namespace std;
vector <int> v[100005];
int n,i,a,b,rad,fiu,vf,k[100005],rez[100005],s[100005],sf[100005];
char viz[100005];
int main()
{
    ifstream f("cerere.in");
    ofstream g("cerere.out");
    f>>n;
    for(i=1; i<=n; i++)
        f>>k[i];
    for(i=1; i<=n-1; i++)
    {
        f>>a>>b;
        viz[b]=1;
        v[a].push_back(b);
    }
    rad=1;
    while(viz[rad]) rad++;
    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<v[rad].size())
        {
            s[++vf]=v[rad][fiu];
            sf[vf]=0;
            sf[vf-1]=fiu+1;
        }
        else vf--;
    }
    for(i=1; i<=n; i++) g<<rez[i]<<' ';
    f.close(); g.close();
    return 0;
}