Problem2095--树上求和

2095: 树上求和

Time Limit: 2 Sec  Memory Limit: 512 MB
Submit: 23  Solved: 4
[Submit] [Status] [Web Board] [Creator:]

Description

给定一颗 n 个节点的树,第 i 个节点有一个颜色 ci

请问对于树上所有点对 u,v(1u<vn),它们路径上的点有多少种不同的颜色。输出所有点对的答案的和。

Input

第一行一个整数 n,表示节点个数。

接下来一行 n 个整数 c1,c2,c3,...,cn 表示每个点的颜色。

接下来 n1 行,每行两个整数 u,v 表示树上的一条边。

Output

一行一个整数,表示答案。

Sample Input Copy

6
1 2 1 3 2 1
1 2
1 3
2 4
2 5
3 6

Sample Output Copy

29

HINT

数据范围:

共 10 组数据

对于测试点 1,2 满足 3n10

对于测试点 3,4 满足 3n100

对于测试点 5,6 满足所有的颜色都相同

对于所有测试点满足 3n5000,1cin

Source/Category