Cod sursa(job #2560212)

Utilizator popashtefan10Popa Stefan popashtefan10 Data 27 februarie 2020 20:28:42
Problema Cerere Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.92 kb
#include <iostream>
#include <cstdio>
#include <vector>
#define NMAX 100000

using namespace std;

int n;
int k[NMAX + 5], ans[NMAX + 5], log2[NMAX + 5], dad[NMAX + 5];
int rmq[17][NMAX + 5];
vector<int> v[NMAX + 5], st;

void dfs(int x, int dx) {
  st.push_back(x);
  if(k[x] == 0)
    ans[x] = 0;
  else
    ans[x] = ans[st[st.size() - k[x] - 1]] + 1;

  for(int y: v[x])
    if(y != dx)
      dfs(y, x);
  st.pop_back();
}

int main() {
  freopen("cerere.in", "r", stdin);
  freopen("cerere.out", "w", stdout);
  int x, y;

  scanf("%d", &n);
  for(int i = 1; i <= n; i++)
    scanf("%d", &k[i]);

  for(int i = 1; i < n; i++) {
    scanf("%d %d", &x, &y);
    dad[y] = x;
    v[x].push_back(y);
    v[y].push_back(x);
  }

  for(int i = 1; i <= n; i++)
    if(dad[i] == 0) {
      dfs(i, 0);
      break;
    }

  for(int i = 1; i <= n; i++)
    printf("%d ", ans[i]);

  return 0;
}