#G0105. 好路径数【2025暑假集训T3】

好路径数【2025暑假集训T3】

题目描述

有一颗 nn 个结点的树,结点 ii 有权值 aia_i

你需要求出这样的点对 (u,v)(u,v) 的数量:

1、au=ava_u = a_v

2、对于 uuvv 的路径上的任意一个点 kk, 都有 akaua_k \geq a_u

点对是无序对,即 (u,v)(u,v)(v,u)(v,u) 是同一种。

输入格式

第一行一个正整数 nn,表示结点数量。

接下来一行 nn 个整数 aia_i,表示结点权值。

接下来 n1n - 1 行每行两个正整数 u,vu,v,表示树上的一条边。

输出格式

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

7
1 2 1 2 1 2 1
1 2
1 3
2 4
3 5
4 6
4 7
9

样例解释

数据规模与约定

下发文件

下发文件分别对应子任务 1155

有合理的子任务依赖。

子任务编号 nn≤ 特殊性质 分值
11 10310^3 ai102a_i \leq 10^2 1010
22 2020
33 3×1053 \times 10^{5} ai102a_i \leq 10^2
44 保证每个点的度数不超过 22
55 3030

对于 100%100\% 的数据:保证 $1 \leq n \leq 3 \times 10^{5},1 \leq u_i \neq v_i \leq n,1 \leq a_i \leq 10^9$。