#B4189. [中山市赛 2024] 树上开花

    ID: 13184 远端评测题 1000ms 512MiB 尝试: 0 已通过: 0 难度: 4 上传者: 标签>二分2024广东最近公共祖先 LCA排列组合小学科创活动

[中山市赛 2024] 树上开花

题目描述

你有一棵以 1 为根的树,统计点对 (x,y)(x, y),满足 alca(x,y)a_{lca(x,y)}axa_xaya_y 的公约数。注意当 xyx \neq y(x,y)(x, y)(y,x)(y, x) 视为不同的点对。

输入格式

第一行一个整数 nn

第二行 nn 个整数 aia_i

第三到 n+1n + 1 行,每行两个整数,表示树上的边。

输出格式

一行一个整数表示答案。

5
2 3 2 5 4
1 2
1 3
2 4
2 5
11

提示

样例解释

以下点对满足条件:(1,1)(1, 1)(1,3)(1, 3)(1,5)(1, 5)(2,2)(2, 2)(3,1)(3, 1)(3,3)(3, 3)(3,5)(3, 5)(4,4)(4, 4)(5,1)(5, 1)(5,3)(5, 3)(5,5)(5, 5)

数据范围

本题数据分为多个子任务,具体如下:

子任务编号 nn 附加条件 子任务分数
11 150\leq 150 1010
22 1500\leq 1500
33 105\leq 10^5 树为随机生成
44 =99998=99998 ai300a_i\leq 300
55 aa1n1\sim n 的排列
66 105\leq 10^5 5050

对于所有数据,保证 1ain1 \leq a_i \leq n