Cod sursa(job #965134)

Utilizator crisbodnarCristian Bodnar crisbodnar Data 23 iunie 2013 14:17:17
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <vector>

#define N 100005

using namespace std;

ifstream fin("cerere.in");
ofstream fout("cerere.out");

int n, m, v, k[N], s[N], poz, sol[N];
vector <int> a[N];
vector <bool> viz(N), t(N);

void dfs(int x)
{
    s[++poz] = x;
    viz[x] = 1;
    if(k[x])
    sol[x] = sol[s[poz - k[x]]] + 1;
    for(unsigned i=0; i<a[x].size(); i++)
        if(!viz[a[x][i]])
            dfs(a[x][i]);
    poz--;
}

int main()
{
    fin>>n;
    int x, y;
    for(int i=1; i<=n; i++)
        fin>>k[i];
    for(int i=1; i<n; i++)
    {
        fin>>x>>y;
        a[x].push_back(y);
        t[y] = 1;
    }
    for(int i=1; i<=n; i++)
        if(!t[i]) {v = i; break;}
    dfs(v);

    for(int i=1; i<=n; i++)
        fout<<sol[i]<<' ';

    return 0;
}