T539630 「YAC Round 12」華鳥風月
题目背景
> 所谓正义之物,即是会随人心之尺而变化
倘若世间存有永恒不变,那必是「華鳥風月」
题目描述
四季映姬·夜摩仙那度,是地狱中制裁死者的最高裁判长,拥有明鉴黑白程度的能力,还能够洞察世间万物的善恶标准。
四季映姬有一颗包含 $n$ 个节点的灵魂之树,每一个节点代表一个灵魂的状态,非黑即白。 为了守护幻想乡冥界的平衡,四季映姬定义了一个“灵魂平衡路径”。她认为树上的一条路径是“灵魂平衡路径”当且仅当同时满足以下两个条件:
- 路径上没有重复经过的节点;
- 路径上相邻节点的灵魂的状态 **均不相同**。
请你求出 **最长**“灵魂平衡路径”的 **长度** 是多少(路径长度定义为路径上节点的数量)。
输入格式
第一行输入一个整数 $n$ ($1 \le n \le 10^5$),表示节点的数量。
第二行输入 $n$ 个整数 $c_1, c_2, c_3, \dots c_n$ ($c_i \in \{ 0, 1\}$,其中 $0$ 代表白,$1$ 代表黑),表示每个节点的灵魂状态。
接下来 $n - 1$ 行,每行两个正整数 $u, v$ ($1 \le u, v \le n$, $u \not = v$),表示节点 $u$ 和 $v$ 之间有一条边。
输出格式
输出一行一个整数,表示最长“灵魂平衡路径”的长度。
说明/提示
### 样例解释
最长的“灵魂平衡路径”为 $1 - 2 - 3 - 5$,共 $4$ 个节点,因此长度为 $4$。