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 翻译