Cod sursa(job #2348004)

Utilizator SqueekDanielTodasca Daniel SqueekDaniel Data 19 februarie 2019 11:56:07
Problema Cerere Scor 65
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.95 kb
#include <bits/stdc++.h>

#define MAXN 100005

int N, Ans[MAXN], Stramos[MAXN];
std::vector <int> ADC[MAXN];

inline void AddEdge(int X, int Y) {
    ADC[X].push_back(Y);
}

int NStack, Stack[MAXN];
void DFS(int Vertex = 1, int Parent = 0) {
    Stack[++NStack] = Vertex;
    if (Stramos[Vertex])
        Ans[Vertex] = Ans[Stack[NStack-Stramos[Vertex]]] + 1;

    for (auto Edge:ADC[Vertex])
        if (Edge != Parent)
            DFS(Edge, Vertex);
    --NStack;
}

std::ifstream In ("cerere.in");
std::ofstream Out("cerere.out");

void Citire() {
    In >> N;
    for (int i=1; i<=N; ++i)
        In >> Stramos[i];
    for (int i=1, X, Y; i<N; ++i)
        In >> X >> Y, AddEdge(X, Y);
}

void Rezolvare() {
    for (int i=1; i<=N; ++i)
        if (!Stramos[i])
            DFS(i), i = N+5;
    for (int i=1; i<=N; ++i, Out << ' ')
        Out << Ans[i];
}

int main()
{
    Citire();
    Rezolvare();

    return 0;
}