Cod sursa(job #3175280)

Utilizator PsyDuck1914Feraru Rares-Serban PsyDuck1914 Data 25 noiembrie 2023 16:27:56
Problema Cerere Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f ("cerere.in");
ofstream g ("cerere.out");

const int NMAX = 1e5;

int k[NMAX+1];
vector<int> adj[NMAX+1];
int lant[NMAX+1];
int rez[NMAX+1];

void dfs(int from, int niv){

    if(k[from] == 0)
        rez[from] = 0;
    else rez[from] = rez[lant[niv - k[from]]] + 1;

    lant[niv] = from;

    for(int to : adj[from]){
        if(rez[to] == 0){
            dfs(to, niv+1);
        }
    }
}

int main()
{
    int n;
    f >> n;

    for(int i=1; i<=n; i++){
        f >> k[i];
    }

    for(int i=1; i<n; i++){
        int x, y;
        f >> x >> y;
        adj[x].push_back(y);
    }

    dfs(1, 0);

    for(int i=1; i<=n; i++)
        g << rez[i] << ' ';

    return 0;
}