【 Codeforces Round #572 (Div. 2)】E. Count Pairs（数学）

题面

题意：

现在有一个长度为$n$的数组$a$，找出有多少对$(i,j)$满足$1\le$$i<j$$\le n$并且$(a_i+a_j)\ast (a_i^2+a_j^2)modp==k$。

思路：

左右同乘$(a_i-a_j)$，则式子化为$(a_i^4-a_j^4)mod$ $p==$ $k\ast (a_i-a_j)$ $mod$ $p$
移项得:
$(a_i^4-k\ast a_i)$ $mod$ $p==$ $(a_j^4-k\ast a_j)$ $mod$ $p$ .
因此我们至于要统计每个$(a_i^4-k\ast a_i)$ $mod$ $p$出现的次数，再累加即可，$map$和$set$可以$O(nlog(n))$的解决这个问题。

#include<bits/stdc++.h>

using namespace std;
set<long long > s;
map<long long ,long long > mmp;
int main(){
	long long n,p,x,k;
	cin>>n>>p>>k;
	for(int i=1;i<=n;i++){
		cin>>x;
		mmp[((x%p*x%p*x%p*x%p)%p-(k%p*x%p)+p)%p]++;
		s.insert(((x%p*x%p*x%p*x%p)%p-(k%p*x%p)%p+p)%p);
	}
	long long ans=0;
	for(long long v:s){
		ans+=(mmp[v]*(mmp[v]-1)/2);
	}
	cout<<ans<<endl;
	return 0;
}