Cod sursa(job #1789660)

Utilizator razvan99hHorhat Razvan razvan99h Data 27 octombrie 2016 12:57:16
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.71 kb
#include <iostream>
#include <fstream>
#include <vector>
#define NM 100005
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n, i, j, x, y, v[NM], viz[NM], rez[NM], st[NM], last, rad;
vector <int> g[NM];

void dfs(int nod)
{
    viz[nod]=1;
    last++;
    st[last]=nod;
    rez[nod]=rez[st[last-v[nod]]]+1;
    for(auto it:g[nod])
        if(!viz[it])
            dfs(it);
    last--;
}
int main()
{
    fin>>n;
    for(i=1;i<=n;i++)
        fin>>v[i];
    for(i=1;i<n;i++)
    {
        fin>>x>>y;
        rad=rad+y;
        g[x].push_back(y);
    }
    rad=n*(n+1)/2-rad;

    dfs(rad);
    for(i=1;i<=n;i++)
        fout<<rez[i]-1<<' ';
    return 0;
}