Cod sursa(job #2359595)

Utilizator MihneaGhiraMihnea MihneaGhira Data 28 februarie 2019 22:23:37
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#include<fstream>
#include<vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,x,y;
int K[100005],f[100005],stiva[100005],D[100005],T[100005];
vector <int> L[100005];

void dfs(int nod,int pas){
    stiva[pas]=nod;
    f[nod]=1;
    D[nod]=1+D[ stiva[pas-K[nod]] ];
    for(int i=0;i<L[nod].size();i++){
        if(f[L[nod][i]]==0){
            dfs(L[nod][i],pas+1);
        }
    }
}

int main(){
    fin>>n;
    for(int i=1;i<=n;i++){
        fin>>K[i];
    }
    for(int i=1;i<n;i++){
        fin>>x>>y;
        L[x].push_back(y);
        L[y].push_back(x);
        T[y]=x;
    }
    for(int i=1;i<n;i++){
        if(T[i]==0){
            dfs(i,0);
            break;
        }
    }
    for(int i=1;i<=n;i++){
        fout<<D[i]-1<<" ";
    }
    return 0;
}