#8305. 「GESP24.06 八级」 最远点对 普及+/提高

时间限制:1000 ms 内存限制:128 MiB 标准输入输出
题目类型:传统 评测方式:文本比较
上传者: root

题目描述

小杨有⼀棵包含 个节点的树,这棵树上的任意⼀个节点要么是白色,要么是黑色。

小杨想知道相距最远的一对不同颜色节点的距离是多少。

输入格式

第一行包含⼀个正整数 ,代表树的节点数。

第二行包含 个非负整数 (对于所有的 ,均有 等于 ),其中如果 ,则节点 的颜色为白色;如果 ,则节点 的颜色为黑色。

之后 行,每行包含两个正整数 ,代表存在一条连接节点 的边。

保证输入的树中存在不同颜色的点。

输出格式

输出⼀个整数,代表相距最远的一对不同颜色节点的距离。

样例

样例输入

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

样例输出

3

样例解释

相距最远的不同颜色的一对节点为节点

数据范围与提示

子任务编号 得分 特殊条件
树的形态为一条链

对于全部数据,保证有