AT_abc198_e [ABC198E] Unique Color

题目描述

给定一棵以 $1$ 节点为根的树。每个节点上都被染了一种颜色,不同颜色用不同整数表示。问:有多少个节点,它们到根节点的路径上的所有点(除了自己)的颜色都与自己的颜色不同。

输入格式

第一行输入整数 $n$($2\le n\le 10^5$)。 第二行输入 $n$ 个整数,第 $i$ 个整数表示节点 $i$ 的颜色。($1\le$ 每个节点的颜色 $\le 10^5$) 最后 $(n-1)$ 行:每行两个数 $a_i$,$b_i$,表示一条树边。

输出格式

一行一个整数,题目所求。

说明/提示

### 制約 - $ 2\ \leq\ N\ \leq\ 10^5 $ - $ 1\ \leq\ C_i\ \leq\ 10^5 $ - $ 1\ \leq\ A_i,B_i\ \leq\ N $ - 与えられるグラフは木である - 入力は全て整数 ### Sample Explanation 1 例えば頂点 $ 1 $ から頂点 $ 6 $ への最短パスには頂点 $ 1,2,3,6 $ が含まれます。これらの中に、頂点 $ 6 $ と同じ色の頂点は頂点 $ 6 $ 以外存在しないので、頂点 $ 6 $ はよい頂点です。 一方で、頂点 $ 1 $ から頂点 $ 5 $ への最短パスには頂点 $ 1,\ 2,\ 5 $ が含まれ、頂点 $ 1 $ と頂点 $ 5 $ の色は同じであるため、頂点 $ 5 $ はよい頂点ではありません。