背景

GESP八级真题(202409)

描述

⼩杨有⼀棵包含 $n$ 个节点的树，节点从 1 到 $n$ 编号，并且每个节点要么是⽩⾊，要么是⿊⾊。

对于树上的⼀条简单路径（不经过重复节点的路径），⼩杨认为它是美丽的当且仅当路径上相邻节点的颜⾊均不相 同。例如下图，其中节点 1 和节点 4 是⿊⾊，其余节点是⽩⾊，路径 2-1-3-4 是美丽路径，⽽ 2-1-3-5 路径 不是美丽路径（相邻节点 3 和 5 颜⾊相同）。

对于树上的⼀条简单路径，⼩杨认为它的长度是路径包含节点的数量。⼩杨想知道最长的美丽路径的长度是多少。

格式

输入

第⼀⾏包含⼀个正整数 $n$ ，代表节点数量。

第⼆⾏包含 $n$ 个整数 $c_1, c_2, ..., c_n$ ，代表每个节点的颜⾊，如果 $c_i = 1$ ，代表节点 $i$ 为⽩⾊，如果 $c+i =0$，代表节点$i$为⿊⾊。

之后 $n-1$ ⾏，每⾏包含两个正整数 $u_i, v_i$ ，代表存在⼀条连接节点 $u_i$ 和节点 $v_i$ 的边。

输出

输出⼀个整数，代表最长美丽路径的长度。

样例

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

4

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

1

数据规模

对于全部数据，保证有 ，同时保证给出的数据构成⼀棵树。 ，

限制

时间限制：1.0 s

空间限制：512.0 MB