Cod sursa(job #953586)

Utilizator BlackLordFMI Alex Oprea BlackLord Data 26 mai 2013 17:49:38
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.71 kb
#include <fstream>
#include <vector>
#define DIM 100010
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int n, i, a[DIM], t[DIM], x, y, b[DIM], N, m, s[DIM];

vector<int> l[DIM];

void dfs(int m){
    b[++N]=m;
    if(a[m])
        s[m]=s[ b[ N-a[m] ] ]+1;
    for(vector<int>::iterator it=l[m].begin(); it!=l[m].end(); it++)
        dfs(*it);
    N--;
}

int main(){
    f>>n;
    for(i=1; i<=n; i++)
        f>>a[i];
    for(i=1; i<n; i++)
    {
        f>>x>>y;
        t[y]=1;
        l[x].push_back(y);
    }
    for(i=1; i<=n; i++)
        if(!t[i])
            break;
    dfs(i);
    for(i=1; i<=n; i++)
        g<<s[i]<<" ";
    g<<"\n";
    return 0;
}