5466 - GESP:2024-6月等级8-T1-最远点对

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

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

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

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

样例解释 相距最远的不同颜色的一对节点为节点2 和5

1<=n<=10^5,0<=ai<=1