A5074 | $午枫的01树中心$

时间限制1s

内存限制256MB

通过 / 提交0/0

题目描述

小枫有一棵 $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$ 树的中心节点个数。

输入 #1

6
1 1 0 0 1 1
1 2
1 3
2 4
3 5
3 6

输出 #1

C++ 编辑器

输入

输出