B4189 [中山市赛 2024] 树上开花
题目描述
你有一棵以 1 为根的树,统计点对 $(x, y)$,满足 $a_{lca(x,y)}$ 是 $a_x$ 和 $a_y$ 的公约数。注意当
$x \neq y$ 时 $(x, y)$ 和 $(y, x)$ 视为不同的点对。
输入格式
第一行一个整数 $n$。
第二行 $n$ 个整数 $a_i$。
第三到 $n + 1$ 行,每行两个整数,表示树上的边。
输出格式
一行一个整数表示答案。
说明/提示
### 样例解释
以下点对满足条件:$(1, 1)$,$(1, 3)$,$(1, 5)$,$(2, 2)$,$(3, 1)$,$(3, 3)$,$(3, 5)$,$(4, 4)$,$(5, 1)$,$(5, 3)$,$(5, 5)$。
### 数据范围
本题数据分为多个子任务,具体如下:
| 子任务编号 | $n$ | 附加条件 | 子任务分数 |
| :----------: | :----------: | :----------: | :----------: |
| $1$ | $\leq 150$ | 无 | $10$ |
| $2$ | $\leq 1500$ | 无 | $10$ |
| $3$ | $\leq 10^5$ | 树为随机生成 | $10$ |
| $4$ | $=99998$ | $a_i\leq 300$ | $10$ |
| $5$ | $=99998$ | $a$ 为 $1\sim n$ 的排列 | $10$ |
| $6$ | $\leq 10^5$ | 无 | $50$ |
对于所有数据,保证 $1 \leq a_i \leq n$。