题目描述

《数据规模》

【GESP202406八级 最远点对】 小杨有一棵包含 n 个节点的树，这棵树上的任意一个节点要么是白色，要么是黑色。小杨想知道相距最远的一对不同颜色节点的距离是多少。

输入格式

第一行包含一个正整数 n，代表树的节点数。第二行包含 n 个非负整数 a1,a2,...,an （对于所有的 1≤i≤n，均有 ai 等于 0 或 1）， 其中如果 ai=0，则节点 i 的颜色为白色；如果 ai=1，则节点 i 的颜色为黑色。 之后 (n-1) 行，每行包含两个正整数 xi,yi，代表存在一条连接节点 xi和 yi的边。 保证输入的树中存在不同颜色的点。

输出格式

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

样例

输入样例 1

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

输出样例 1

1

输入样例 2

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

输出样例 2

3

输入样例 3

7
0 0 1 0 1 0 1
1 2
2 3
1 4
4 5
1 6
6 7

输出样例 3

3