Cod sursa(job #1306831)

Utilizator ovidiuz98Zamfir Ovidiu ovidiuz98 Data 31 decembrie 2014 16:08:00
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.75 kb
#include <fstream>
#include <vector>
#define DIM 100002
using namespace std;

ifstream fin("cerere.in");
ofstream fout("cerere.out");
vector <int> v[DIM];
int n,s[DIM],t[DIM],sol[DIM],nv[DIM],viz[DIM],niv;
void DFS(int x){
    nv[++niv]=x;
    if(s[x]!=0){
        sol[x]=sol[ nv[ niv - s[x] ] ]+1;
    }
    for(int i=0;i<v[x].size();i++)
            DFS(v[x][i]);
    niv--;
}
int main(){

    fin>>n;
    for(int i=1;i<=n;i++){
        fin>>s[i];
    }
    for(int i=2;i<=n;i++){
        int x,y;
        fin>>x>>y;
        v[x].push_back(y);
        t[y]=x;
    }
    for(int i=1;i<=n;i++)
        if(!t[i])
            DFS(i);
    for(int i=1;i<=n;i++)
        fout<<sol[i]<<" ";
    fin.close();fout.close();
    return 0;
}