2023661 - gesp-2024-6-08-1寻找倍数

题目描述

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

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

输入

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

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

输出

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

样例

输入

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

输出

3
说明

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

对于全部数据,保证有 1≤n≤105,0≤ai≤1

题目参数
时间限制 1 秒
内存限制 128 MB
提交次数 1
通过人数 1
金币数量 1 枚
统计
上一题 下一题