给定一颗 n 个节点的树,第 i 个节点有一个颜色 ci。
请问对于树上所有点对 u,v(1≤u<v≤n),它们路径上的点有多少种不同的颜色。输出所有点对的答案的和。
第一行一个整数 n,表示节点个数。
接下来一行 n 个整数 c1,c2,c3,...,cn 表示每个点的颜色。
接下来 n−1 行,每行两个整数 u,v 表示树上的一条边。
6
1 2 1 3 2 1
1 2
1 3
2 4
2 5
3 6
29
数据范围:
共 10 组数据
对于测试点 1,2 满足 3≤n≤10
对于测试点 3,4 满足 3≤n≤100
对于测试点 5,6 满足所有的颜色都相同
对于所有测试点满足 3≤n≤5000,1≤ci≤n