A12706 | Count Pairs
时间限制1s
内存限制256MB
通过 / 提交0/0
题目描述
You are given a prime number $p$ , $n$ integers $a_1, a_2, \ldots, a_n$ , and an integer $k$ .
Find the number of pairs of indexes $(i, j)$ ( $1 \le i < j \le n$ ) for which $(a_i + a_j)(a_i^2 + a_j^2) \equiv k \bmod p$ .
Find the number of pairs of indexes $(i, j)$ ( $1 \le i < j \le n$ ) for which $(a_i + a_j)(a_i^2 + a_j^2) \equiv k \bmod p$ .
输入格式
The first line contains integers $n, p, k$ ( $2 \le n \le 3 \cdot 10^5$ , $2 \le p \le 10^9$ , $0 \le k \le p-1$ ). $p$ is guaranteed to be prime.
The second line contains $n$ integers $a_1, a_2, \ldots, a_n$ ( $0 \le a_i \le p-1$ ). It is guaranteed that all elements are different.
The second line contains $n$ integers $a_1, a_2, \ldots, a_n$ ( $0 \le a_i \le p-1$ ). It is guaranteed that all elements are different.
输出格式
Output a single integer — answer to the problem.
输入输出样例
输入 #1
3 3 0 0 1 2
输出 #1
1
输入 #2
6 7 2 1 2 3 4 5 6
输出 #2
3
In the first example:
$(0+1)(0^2 + 1^2) = 1 \equiv 1 \bmod 3$ .
$(0+2)(0^2 + 2^2) = 8 \equiv 2 \bmod 3$ .
$(1+2)(1^2 + 2^2) = 15 \equiv 0 \bmod 3$ .
So only $1$ pair satisfies the condition.
In the second example, there are $3$ such pairs: $(1, 5)$ , $(2, 3)$ , $(4, 6)$ .
$(0+1)(0^2 + 1^2) = 1 \equiv 1 \bmod 3$ .
$(0+2)(0^2 + 2^2) = 8 \equiv 2 \bmod 3$ .
$(1+2)(1^2 + 2^2) = 15 \equiv 0 \bmod 3$ .
So only $1$ pair satisfies the condition.
In the second example, there are $3$ such pairs: $(1, 5)$ , $(2, 3)$ , $(4, 6)$ .
C++ 编辑器
输入
输出
可保存默认模板；新题优先使用已保存模板。
当前快捷键仅展示，暂不支持修改。
- 撤销
Ctrl / ⌘ + Z - 重做
Ctrl / ⌘ + Y - 查找
Ctrl / ⌘ + F - 全选
Ctrl / ⌘ + A - 复制
Ctrl / ⌘ + C - 剪切
Ctrl / ⌘ + X - 粘贴
Ctrl / ⌘ + V - 自动排版
工具栏排版按钮 - 草稿保存
编辑时自动保存到本机
历史
提交记录
状态说明时间源码
AI
作答助手
你好，我是作答助手。可以问思路、复杂度、样例含义或代码报错原因；不会直接给出完整 AC 代码。
确定要清空代码吗？
提交通过
评测结果：Accepted