Codeforces Round #572 (Div. 2)E. Count Pairs（数学+思维）

题目链接：点这里

/***************************
Codeforces Round #572 (Div. 2)E. Count Pairs（数学+思维） 

题目链接：http://codeforces.com/contest/1189/problem/E

题目大意：给你n,p,k以及n个数求出(ai+aj)(ai^2+aj^2) ≡k mod p ;其中ai和aj存在的对数

题目思路：n有3e5，显然暴力是不可能的，这是一个数学题，将方程左右两边同乘一个ai-aj，
就会变成ai^4-aj^4≡k(ai-aj),则ai^4-k*ai≡aj^4-k*aj mod p,即左右两式取模p始终相等，
那么我们就可以预处理除取模后的结果，然后用map记录出现了多少次，就能算出他们的匹配对数 
 
****************************/
#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll a[300005];
int main()
{
	ll n,p,k,ans=0;
	cin>>n>>p>>k;
	map<ll,ll>mp;
	for(ll i=0;i<n;i++){
		scanf("%lld",&a[i]);
		a[i]=((((a[i]*a[i])%p)*((a[i]*a[i])%p))%p-(k*a[i])%p+p)%p;
		mp[a[i]]++;
	}
	for(ll i=0;i<n;i++){
		if(mp[a[i]]) ans+=(mp[a[i]]*(mp[a[i]]-1))/2,mp[a[i]]=0;
	}
	cout<<ans<<endl;
	return 0;
}