#P1006. [2024 校赛] 神秘三人组

[2024 校赛] 神秘三人组

题目描述

本题题解已发表至 讨论区

题面背景

在遥远的未来,科技与魔法交织的时代,ACM赛事不仅是编程精英们展示智慧与技巧的舞台,更是探索未知、解决世界难题的象征。这项赛事之所以采用三人组队的形式,背后蕴含着深远的寓意——正如古老的几何学智慧所揭示,三角形是自然界中最稳定的结构之一。它代表着均衡、协作与力量的完美结合,象征着在复杂多变的环境中,三个智慧生命的集合能够激发出超越个体的非凡能力。

在这个故事里,神秘三人组并非普通的编程高手,他们是来自不同领域的精英:一位精通量子计算的物理天才,一位擅长人工智能与机器学习的数据科学家,以及一位拥有超凡直觉与创造力的软件工程师。他们因一次偶然的机会,在ACM赛事中相遇,并凭借无与伦比的默契与实力,屡创佳绩,成为了众人眼中的传奇。

然而,命运似乎总爱开玩笑。正当三人准备在下一届ACM赛事中大放异彩之时,一场突如其来的灾难打破了平静。一块蕴含未知能量的陨石,不偏不倚地坠落在了他们共同工作的实验室上方,其内部的不稳定物质正悄然酝酿着一场毁灭性的爆炸。时间紧迫,留给他们的只有短短几个小时,来解开陨石的秘密,阻止这场浩劫。

为何偏偏是三人?这不仅是因为三角形稳定的象征意义,更是因为在这紧急关头,三人各自的专业技能与思维方式构成了解决问题的完美闭环。物理天才利用对量子理论的深刻理解,尝试分析陨石内部能量的波动规律;数据科学家则运用大数据分析与机器学习算法,预测爆炸的可能时间与规模;而软件工程师,则利用自己的创造力,设计出一套能够引导并中和陨石能量的智能系统。

在紧张的氛围中,三人分工明确,却又紧密相连。他们之间的每一次交流,都闪烁着智慧与信任的火花,仿佛那无形的三角形纽带,将他们紧紧绑在一起,共同面对挑战。经过无数次的尝试与修正,终于,在最后一刻,他们成功激活了智能系统,将陨石内部的不稳定能量引导至安全区域释放,避免了一场灾难的发生。

这次经历,不仅加深了三人之间的友谊,更让他们意识到,在这个充满未知与危险的世界里,真正的力量来自于团队的紧密合作与相互信任。从此,神秘三人组不仅成为了ACM赛事的传奇,更成为了守护世界安宁的英雄,他们的故事激励着后来者,证明了在智慧与勇气的照耀下,即使是再大的困难,也无法阻挡三人携手前行的脚步。

现在给定你n个实验室成员的能力值和匹配值p,请问你可以选出多少种神秘三人组的组队方式出来。现在给定你n个实验室成员的能力值和匹配值p,请问你可以选出多少种神秘三人组的组队方式出来。

给定三个人的能力值他们可以组队当前仅当符合以下要求给定三个人的能力值他们可以组队当前仅当符合以下要求

假设选择了三个人他们的下标分别是i,j,k(0<i<j<k<=n)假设选择了三个人他们的下标分别是i,j,k (0<i<j<k<=n)
则需要满以下两个条件(注意是数学中的严格相等,不存在向下取整的情况)则需要满以下两个条件(注意是数学中的严格相等,不存在向下取整的情况)

1.aip=aj1.a_i*p= a_j
2.ai=ak/p22. a_i = a_k /p^2

输入描述

第一行两个数字nk分别表示实验室成员数量,匹配值k第一行两个数字n,k分别表示实验室成员数量,匹配值k

1<=n<=2105,1<=p<=109(1<=n<=2*10^5,1<=p<=10^9)

第二行输入n个数字,表示从a1an表示对应成员的能力值其中第二行输入n个数字,表示从a_1到a_n表示对应成员的能力值其中

(0<=ai<=109)(0<=a_i<=10^9)

输出描述

输出一个数表示可以组队的数量输出一个数表示可以组队的数量

示例 1

输入

5 2
1 1 2 2 4

输出

4

示例 2

输入

3 1
1 1 1

输出

1

示例 3

输入

10 3
1 2 6 2 3 6 9 18 3 9

输出

6