Cod sursa(job #2542082)

Utilizator CiboAndreiAndrei Cibo CiboAndrei Data 9 februarie 2020 14:19:58
Problema Cerere Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.01 kb
#include <bits/stdc++.h>

using namespace std;

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

const int dim = 100001;

int n;
int k[dim];

vector <int> sons[dim];
int parent[dim];
int dist[dim];

bool mark[dim];

void dfs(int node){
    mark[node] = 1;

    for(int it: sons[node]){
        if(!mark[it]){
            if(k[it] > 0){
                int p = it;
                for(int i = 1; i <= k[it]; ++i)
                    p = parent[p];
                dist[it] = dist[p] + 1;
            }

            dfs(it);
        }
    }
}

int main()
{
    int i, j, a, b;

    f >> n;

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

    for(i = 1; i < n; ++i){
        f >> a >> b;
        sons[a].push_back(b);
        parent[b] = a;
    }

    int root;

    for(i = 1; i <= n; ++i)
        if(parent[i] == 0){
            root = i;
            break;
        }

    dfs(root);

    for(i = 1; i <= n; ++i)
        g << dist[i] << " ";

    return 0;
}