A. 徐老师的等式求解

    传统题 2000ms 512MiB

徐老师的等式求解

该比赛已结束,您无法在比赛模式下递交该题目。您可以点击“在题库中打开”以普通模式查看和递交本题。

题目描述

徐老师最近跟石老师进行了数学的地狱训练

今天他碰到了一个难题!

石老师给徐老师随机写了一个长度为 nn 的序列 a1∼ana_1 \sim a_n 和一个质数 pp

让徐老师求有多少个数对 (i,j)(i, j) 满足:

  1. 1≤i,j≤n1 \leq i, j \leq n
  2. i≠ji \neq j
  3. ai2+ai+aj2≡ai×aj(modp)a^2_i + a_i + a^2_j \equiv a_i \times a_j \pmod{p}。

现在徐老师被这个问题难住了,请你帮他计算一下

P.S. 你可以将 A≡B(modp)A \equiv B \pmod{p} 简单的理解为 A mod p=B mod pA\ mod\ p = B\ mod\ p

输入格式

输入第一行为两个正整数 n,pn, p,含义如题

输入第二行为 nn 个正整数,分别表示 a1∼ana_1 \sim a_n

输出格式

输出一个整数,表示满足条件的数对个数。

数据范围

对于全部测试点满足:$1 \leq n \leq 5 \times 10^5, 1 \leq p \leq 10^9 + 30$, $1 \leq a_i < \left\lfloor \frac{p}{2} \right\rfloor$,且保证 pp 是质数。

测试点编号 特殊性质
1∼21 \sim 2 n≤3000n \leq 3000
3∼43 \sim 4 p≤7000p \leq 7000
5∼105 \sim 10 无

样例输入1

6 17
6 3 5 2 4 1

样例输出1

4

样例解释1

数对 (1,4),(1,5),(5,2),(5,6)(1, 4), (1, 5),(5, 2), (5, 6) 满足条件。

2026暑假提高组模拟赛(6)

未参加
状态
已结束
规则
IOI
题目
4
开始于
2026-8-5 17:30
结束于
2026-8-15 17:30
持续时间
240 小时
主持人
参赛人数
21