该比赛已结束，您无法在比赛模式下递交该题目。您可以点击“在题库中打开”以普通模式查看和递交本题。

3.1.1题面描述
小杨有一棵包含n个节点的树，这棵树上的任意一个节点要么是白色，要么是黑色。
小杨想知道相距最远的一对不同颜色节点的距离是多少。
3.1.2输入格式
第一行包含一个正整数n,代表树的节点数。
第二行包含n个非负整数a1,a2....,an (对于所有的1≤i≤n， 均有ai等于0或1)，其中如果ai=0,则节点i的颜色为白色;如果ai=1，则节点i的颜色为黑色。
之后n-1行，每行包含两个正整数xi,yi, 代表存在一条连接节点xi和Yi的边。
保证输入的树中存在不同颜色的点。
3.1.3输出格式
输出一个整数，代表相距最远的一对不同颜色节点的距离。
3.1.4样例1
输入：
5
0 1 0 1 0
1 2
1 3
3 4
3 5
输出：
3
3.1.5样例解释
相距最远的不同颜色的一对节点为节点2和5。
3.1.6数据范围
对于全部数据，保证有1≤n≤10^5, 0≤ai≤1。