A59006.午枫的01树中心

小枫有一棵 $n$ 个节点 $01$ 树，每个节点的权值是 $0$ 或 $1$ ，定义中心节点 $x$ ，所有与节点 $x$ 距离为 $1$ 的节点的权值都与节点 $x$ 的权值不同。

小午想知道这颗 $01$ 树有多少个中心节点。

第一行输入一个正整数 $n$ $(2\leq n\leq 2\times 10^5)$ ，表示节点个数。

第二行输入 $n$ 个整数 $a_i$ $(a_i\in\{0,1\})$ ，表示第 $i$ 个节点权值。

接下来 $n-1$ 行，每行输入两个正整数 $u,v$ $(1\leq u,v\leq n)$ ，表示 $u$ 和 $v$ 之间有一条边连接。

输出一个整数，表示这颗 $01$ 树的中心节点个数。

这颗 $01$ 树的 $4$ 个中心节点为 $3$，$4$，$5$，$6$ 。