#G0105. 好路径数【2025暑假集训T3】
好路径数【2025暑假集训T3】
题目描述
有一颗 个结点的树,结点 有权值 。
你需要求出这样的点对 的数量:
1、
2、对于 到 的路径上的任意一个点 , 都有 。
点对是无序对,即 和 是同一种。
输入格式
第一行一个正整数 ,表示结点数量。
接下来一行 个整数 ,表示结点权值。
接下来 行每行两个正整数 ,表示树上的一条边。
输出格式
一行一个非负整数,表示答案。
7
1 2 1 2 1 2 1
1 2
1 3
2 4
3 5
4 6
4 7
9
样例解释
无
数据规模与约定
下发文件分别对应子任务 、。
有合理的子任务依赖。
子任务编号 | 特殊性质 | 分值 | |
---|---|---|---|
保证每个点的度数不超过 | |||
对于 的数据:保证 $1 \leq n \leq 3 \times 10^{5},1 \leq u_i \neq v_i \leq n,1 \leq a_i \leq 10^9$。