AT_nupc2024_e Exponential

题目描述

给定一个长度为 $N$ 的整数序列 $A = (A_1, A_2, \dots, A_N)$ 和一个整数 $M$。 请计算满足以下所有条件的整数三元组 $(i, j, k)$ 的个数。 - $1 \leq i, j, k \leq N$ - $i, j, k$ 互不相同 - $A_i \times M^{A_j} = A_k$

输入格式

输入为以下格式,通过标准输入给出。 > $N$ $M$ $A_1$ $A_2$ $\dots$ $A_N$

输出格式

输出答案。

说明/提示

## 样例解释 1 满足条件的 $(i, j, k)$ 的组合如下,共有 $6$ 种。 - $(1, 2, 3)$ - $(1, 3, 5)$ - $(2, 1, 3)$ - $(3, 1, 4)$ - $(3, 2, 5)$ - $(4, 1, 5)$ ## 数据范围 - $3 \leq N \leq 2 \times 10^5$ - $1 \leq M \leq 10^{18}$ - $0 \leq A_i \leq 10^{18}$ - 输入均为整数。 由 ChatGPT 5 翻译