g6-07640. 最长同色路径 未做

给定一棵 $n$ 个结点的树，每个结点颜色为 $0$ 或 $1$。求一条路径上所有结点颜色相同的简单路径的最大长度（点数）。

输入格式：第一行 $n$；第二行 $n$ 个颜色；接下来 $n-1$ 行 $u,v$。

输入：

4
0 0 0 1
1 2
1 3
2 4

输出：

3

路径 $3-1-2$ 三个点同色 $0$。

数据范围：$1\le n\le 2\times 10^5$。