CF1467E Distinctive Roots in a Tree

给定一棵有 $n$ 个顶点的树。每个顶点 $i$ 都有一个与之关联的值 $a_i$。 我们可以选择某个顶点 $v$ 作为树的根。如果满足以下条件，则称顶点 $v$ 是一个“特殊根”：对于所有从 $v$ 出发到达其它任意节点的路径，路径上经过的所有值都互不相同。不同的路径可以有相同的值，但同一条路径上所有值必须互不相同。 请你求出树中有多少个“特殊根”。

第一行包含一个整数 $n$（$1 \le n \le 2\cdot10^5$），表示树的顶点数。 第二行包含 $n$ 个用空格分隔的整数 $a_1, a_2, \dots, a_n$（$1 \le a_i \le 10^9$），表示每个顶点的值。 接下来的 $n-1$ 行，每行包含两个用空格分隔的整数 $u$ 和 $v$（$1 \le u, v \le n$），表示一条连接 $u$ 和 $v$ 的边。 保证这些边构成一棵树。

输出一个整数，表示树中“特殊根”的数量。

在第一个样例中，$1$、$2$ 和 $5$ 都是“特殊根”。 由 ChatGPT 4.1 翻译