6 lines
629 B
TeX
6 lines
629 B
TeX
A primeira linha da entrada contém dois inteiros $n$ e $m$ ($2 \le n \le 10^5, 1 \le m \le n$), indicando o número de salas no abrigo e o número máximo tolerado de salas consecutivas com gatinhos.
|
|
|
|
A segunda linha contém $n$ inteiros $a_1, a_2, \ldots, a_n$ ($a_i \in \{0, 1\}$). Se $a_i = 1$, a sala $i$ tem gatinhos. Se $a_i = 0$, a sala $i$ não tem gatinhos.
|
|
|
|
As próximas $n-1$ linhas descrevem os corredores do abrigo. Cada linha contém dois inteiros $u$ e $v$ ($1 \le u, v \le n, u \neq v$), indicando um corredor entre as salas $u$ e $v$. É garantido que os corredores formam uma árvore válida enraizada em $1$.
|