U443380 [GESP202406八级]

GESP 2024年6月认证 八级编程题 T2

一棵由 $n$ 个节点组成的树，第 $i$ 个节点的编号为 $i$。对于每个节点 $i$，都有一个颜色，用整数表示为 $a_i$。请求出颜色不同的最远节点之间的距离。

$n+2$ 行。 第 $1$ 行，一个整数 $n$，表示树中的节点个数。 第 $2$ 行，$n$ 个整数，第 $i$ 个整数 $a_i$ 表示编号为 $i$ 的节点的颜色。 第 $3$ 至 $n+2$ 行，每行 $2$ 个整数。第 $i+2$ 行两个整数 $x_i,y_i$ 表示节点 $x_i$ 与 $y_i$ 间联一条边。

$1$ 行，$1$ 个整数，表示颜色不同的最远节点之间的距离。

样例解释： 最远的两个节点是 $2$ 和 $5$。 数据范围： 对于前 $30%$ 的测试点，保证 $1 \leq n \leq 10^5$，树构成一条链。 对于另外 $30%$ 的测试点，保证 $1 \leq n \leq 10^3$。 对于 $100%$ 的测试点，保证$1 \leq n \leq 10^5,0 \leq a_i \leq 1$。