Cod sursa(job #2835528)

Utilizator LionMan101Achim-Panescu Silvian LionMan101 Data 18 ianuarie 2022 20:25:00
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.46 kb
#include <bits/stdc++.h>

using namespace std;

using ll = long long;
const int INF=1e9+7;
const ll INFF=1e18+7;
#define nl '\n'
void __print(int x) { cout << x; }
void __print(long x) { cout << x; }
void __print(long long x) { cout << x; }
void __print(unsigned x) { cout << x; }
void __print(unsigned long x) { cout << x; }
void __print(unsigned long long x) { cout << x; }
void __print(float x) { cout << x; }
void __print(double x) { cout << x; }
void __print(long double x) { cout << x; }
void __print(char x) { cout << '\'' << x << '\''; }
void __print(const char* x) { cout << '\"' << x << '\"'; }
void __print(const string& x) { cout << '\"' << x << '\"'; }
void __print(bool x) { cout << (x ? "true" : "false"); }
template<typename T, typename V>
void __print(const pair<T, V>& x) { cout << '{'; __print(x.first); cout << ','; __print(x.second); cout << '}'; }
template<typename T>
void __print(const T& x) { int f = 0; cout << '{'; for (auto& i : x) cout << (f++ ? "," : ""), __print(i); cout << "}"; }
void _print() { cout << "]\n"; }
template <typename T, typename... V>
void _print(T t, V... v) { __print(t); if (sizeof...(v)) cout << ", "; _print(v...); }
#ifndef ONLINE_JUDGE
#define debug(x...) cout << "[" << #x << "] = ["; _print(x)
#else
#define debug(x...)
#endif

int n,root;
vector<vector<int>> a(100001);
vector<int> k(100001),parent(100001),depth(100001), ans(100001);
vector<bool> visited(100001);

void dfs(int x, int l){
    depth[l]=x;
    visited[x]=true;
    if(k[x]){
        ans[x]=1+ans[depth[l-k[x]]];
    }
    for(auto&i:a[x]){
        if(!visited[i]){
            dfs(i,l+1);
        }
    }
}


void solve()
{
    cin>>n;
    for(int i=1; i<=n; i++){
        cin>>k[i];
    }
    for(int i=1; i<n; i++){
        int x,y;
        cin>>x>>y;
        a[x].emplace_back(y);
        parent[y]=x;
    }
    for(int i=1; i<=n; i++){
        if(!parent[i]){
            root=i;
            break;
        }
    }
    dfs(root,0);
    for(int i=1; i<=n; i++){
        cout<<ans[i]<<" ";
    }
}

int main()
{
// #ifndef ONLINE_JUDGE
    // freopen("input.txt", "r", stdin);
    // freopen("output.txt", "w", stdout);
// #endif
    freopen("cerere.in", "r", stdin);
    freopen("cerere.out", "w", stdout);
    ios::sync_with_stdio(0);
    cin.tie(0);
    int t=1;
    // int t;
    // cin >> t;
    for(int tt=1; tt<=t; tt++){
        // cout<<"#Case "<<t<<nl;
        solve();
    }
    return 0;
}