1

KMODP
Cho một số nguyên tố 𝑝, 𝑛 số nguyên 𝑎1 , 𝑎2 , … , 𝑎𝑛 và một số nguyên 𝑘.
Yêu cầu: Đếm số lượng cặp (𝑖, 𝑗)(1 ≤ 𝑖 < 𝑗 ≤ 𝑛) thỏa mãn:
(𝑎𝑖 + 𝑎𝑗 ) ∗ (𝑎𝑖2 + 𝑎𝑗2 ) ≡ 𝑘 𝑚𝑜𝑑 𝑝
Input:
• Dòng đầu tiên chứa ba số 𝑛, 𝑘, 𝑝(2 ≤ 𝑛 ≤ 3 ∗ 105 , 2 ≤ 𝑝 ≤ 109 , 0 ≤ 𝑘 ≤ 𝑝 − 1)
• Dòng thứ hai chứa 𝑛 số nguyên 𝑎1 , 𝑎2 , … , 𝑎𝑛 (0 ≤ 𝑎𝑖 ≤ 𝑝 − 1). Đảm bảo rằng 𝑛
phần tử trên phân biệt.

Output: In ra một dòng là số lượng cặp thỏa mãn.

Ví dụ:
Sample Input Sample Output
330 1
012
672 3
123456

Giải thích ví dụ: Ở ví dụ đầu tiên:

• (0 + 1) ∗ (02 + 12 ) = 1 ≡ 1 𝑚𝑜𝑑 3
• (0 + 2) ∗ (02 + 22 ) = 8 ≡ 2 𝑚𝑜𝑑 3
• (1 + 2) ∗ (12 + 22 ) = 15 ≡ 0 𝑚𝑜𝑑 3

Vậy chỉ có một cặp thỏa mãn

LEARNING IS THE EYE OF THE MIND