Cod sursa(job #2358688)

Utilizator MihneaGhiraMihnea MihneaGhira Data 28 februarie 2019 11:20:15
Problema Cerere Scor 65
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.8 kb
#include<fstream>
#include<vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,x,y;
int stiva[100001],D[100001],v[100001],K[100001];
vector <int> L[100001];

void dfs(int nod,int pas){
    stiva[pas]=nod;
    D[nod]=1+D[stiva[pas-K[nod]]];
    v[nod]=1;
    for(int i=0;i<L[nod].size();i++){
        if(v[L[nod][i]]==0){
            v[L[nod][i]]=1;
            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-1;i++){
        fin>>x>>y;
        L[x].push_back(y);
        L[y].push_back(x);
    }
    for(int i=1;i<=n;i++)
        if(K[i]==0){
            dfs(i,0);
            break;
        }
    for(int i=1;i<=n;i++)
        fout<<D[i]-1<<" ";
    return 0;
}