Cod sursa(job #2560247)

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

using namespace std;

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

void dfs(int x) {
  st[++vf] = x;
  if(k[x] == 0)
    ans[x] = 0;
  else
    ans[x] = ans[st[vf - k[x]]] + 1;

  for(int i = 0; i < v[x].size(); i++)
    dfs(v[x][i]);
  --vf;
}

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);
  }

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

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

  return 0;
}