#P13875. [蓝桥杯 2024 省研究生组] 植物生命力

[蓝桥杯 2024 省研究生组] 植物生命力

题目描述

小蓝是一位资深的植物学家。他专注于研究植物的相互关系和生命力。在他所照料的森林中,每个品种的植物都拥有独特的生命力,彼此之间互不相同。

植物的生命力会影响其下级品种的生长。具体地,如果下级品种的生命力数值无法被上级品种的生命力数值整除,或者下级品种的生命力数值大于上级品种的生命力数值时,它们便会受到压制,无法茁壮成长。

为了深入研究和定量分析这一现象,小蓝构建了一种模型。他将森林中的植物品种关系抽象成了一棵包含 nn 个结点的树,结点的编号从 1 到 nn,代表不同的植物品种。其中,树的根结点编号为 ss,结点 ii1in1 \leq i \leq n)的生命力表示为 aia_i

现在,小蓝想要对于每个结点 ii,统计其子树(以 ii 为根的子树)中同时满足以下两个条件的子结点的数量:

  1. 子结点的生命力小于结点 ii 的生命力 aia_i
  2. 子结点的生命力无法被结点 ii 的生命力 aia_i 整除。

请你帮助小蓝计算出所有子树中满足条件的结点个数的总和。

输入格式

输入的第一行包含两个整数 nnss,分别表示结点的数量和根结点的编号。

第二行包含 nn 个互不相同的整数 a1,a2,,ana_1, a_2, \cdots, a_n,相邻整数之间使用一个空格分隔,其中 aia_i 表示编号为 ii 的结点的生命力。

接下来的 n1n-1 行,每行包含两个整数 uuvv,用一个空格分隔,表示编号为 uuvv 的结点之间存在一条边。

输出格式

输出一行包含一个整数,表示所有子树中满足条件的结点个数的总和。

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

提示

【样例说明】

在给定的样例中,树的结构如下:

    1
   / \
  2   3
 / \   \
4  5    6

在以 11 为根的子树中,满足条件的结点有 2,52, 5,个数为 22

在以 22 为根的子树中,满足条件的结点有 4,54, 5,个数为 22

在以 363 \sim 6 为根的子树中,没有满足条件的结点,个数均为 00

因此答案为 2+2=42 + 2 = 4

【评测用例规模与约定】

对于 30%30\% 的评测用例,1n2×1031 \leq n \leq 2 \times 10^31s,u,v,ain1 \leq s, u, v, a_i \leq na1,a2,,ana_1, a_2, \ldots, a_n 互不相同。

对于所有评测用例,1n1051 \leq n \leq 10^51s,u,v,ain1 \leq s, u, v, a_i \leq na1,a2,,ana_1, a_2, \ldots, a_n 互不相同。