王国
题号:NC205899
时间限制:C/C++/Rust/Pascal 1秒,其他语言2秒
空间限制:C/C++/Rust/Pascal 128 M,其他语言256 M
64bit IO Format: %lld

题目描述

这个世界上有个王国,有n-1条道路,任意两个王国相互可达. 每个王国都有属于某个势力,王国i属于势力 ,势力相同的王国想要相互联络.王国a与王国b的联络难度为(x为a到b简单路径上边的数量)

输出最大的联络难度.

输入描述:

第一行为正数n,表示王国数量

第二行为n个正整数,a_1,a_2...a_n ,a_i 表示王国i所属的势力.

接下来的n-1行,每行2个正整数u,v. 表示王国u与王国v有道路相连.

输出描述:

输出最大联络难度
示例1

输入

复制
7
1 1 2 3 3 3 1
1 2
1 3
2 4
2 5
3 6
1 7

输出

复制
16

说明

王国4与王国6联系的难度为16

备注: